from transformers import T5ForConditionalGeneration, TFT5ForConditionalGeneration | |
pt_model = T5ForConditionalGeneration.from_pretrained(".", from_flax=True) | |
pt_model.save_pretrained(".") | |
tf_model = TFT5ForConditionalGeneration.from_pretrained(".", from_pt=True) | |
tf_model.save_pretrained(".") | |