flax-community
/

gpt2-medium-indonesian

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

gpt2-medium-indonesian / jax2torch.py

cahya's picture

add config

aa4d81d over 3 years ago

210 Bytes

	from transformers import GPT2Config, GPT2LMHeadModel

	'''
	This is a script to convert the Jax model to Pytorch model
	'''

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