birgermoell
/

swedish-gpt

Text Generation Transformers PyTorch JAX TensorBoard Swedish gpt2 Inference Endpoints text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

swedish-gpt / save_model.py

birgermoell's picture

Added model

8a454b5 almost 3 years ago

raw history blame contribute delete

No virus

306 Bytes

	from transformers import AutoTokenizer, GPT2LMHeadModel
	'''
	This is a script to convert the Jax model and the tokenizer to Pytorch model
	'''
	model = GPT2LMHeadModel.from_pretrained(".", from_flax=True)
	model.save_pretrained(".")
	tokenizer = AutoTokenizer.from_pretrained(".")
	tokenizer.save_pretrained(".")