flax-community
/

norsk-gpt-wiki

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

Model card Files Files and versions Metrics Training metrics Community

norsk-gpt-wiki / save_model.py

birgermoell's picture

Trained model upload

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