flax-community
/

gpt2-swahili

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

gpt2-swahili / flax_to_torch.py

alokmatta

gpt2 swahili

49a1166 over 3 years ago

history blame contribute delete

131 Bytes

	from transformers import GPT2LMHeadModel

	model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
	model.save_pretrained("./")