flax-community
/

gpt2-medium-indonesian

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

gpt2-medium-indonesian / jax2torch.py

cahya's picture

add config

aa4d81d almost 3 years ago

raw history blame

No virus

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(".")