Bengali-t5 / prev-checkpoint /convert_flax_model_to_pytorch.py
ibraheemmoosa's picture
Pytorch checkpoint of last run.
062d2af
raw
history blame contribute delete
164 Bytes
from transformers import T5ForConditionalGeneration
model = T5ForConditionalGeneration.from_pretrained('.', from_flax=True)
model.save_pretrained('pytorch_model')