from transformers import T5ForConditionalGeneration

pt_model = T5ForConditionalGeneration.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".")