Pre-training googlebyt5small

Is it possible to use the run_t5_mlm_flax.py script (link below) in order to pre-train the googlebyt5small and google/t5-small-lm-adapt models. Or there is a different script in order to do so?

Thanks in advanceā€¦