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