gpt-2-tamil / src /convert_flax_to_pytorch.py
AbinayaM02
Converted flax to pytorch model
ed4e9b8
raw
history blame
145 Bytes
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("../gpt-2-tamil", from_flax=True)
model.save_pretrained("../")