t5-base-dutch / flax_to_pt.py
yhavinga's picture
Saving weights and logs of step 600
d97a524
raw
history blame
158 Bytes
from transformers import T5ForConditionalGeneration
pt_model = T5ForConditionalGeneration.from_pretrained(".", from_flax=True)
pt_model.save_pretrained(".")