If you did a fresh pull of that fork using a Colab notebook that does so, it will set the flag automatically if you specify the 345M model (I will follow that behavior in my own package).

If it didn't set the flag, it gets messy: https://twitter.com/minimaxir/status/1124742105421631488