/home/bayartsogtyadamsuren/transformers/examples/flax/language-modeling/run_clm_flax.py