flax-community
/

gpt-2-tamil

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

gpt-2-tamil / src /convert_flax_to_pytorch.py

Abinaya Mahendiran

Convertd flax to pytorch model

f782d18 almost 3 years ago

raw history blame contribute delete

No virus

145 Bytes

	from transformers import GPT2LMHeadModel

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