Files
gpt-2-tamil/src/convert_flax_to_pytorch.py

5 lines
145 B
Python
Raw Normal View History

from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("../gpt-2-tamil", from_flax=True)
model.save_pretrained("../")