flax-community
/

gpt2-base-thai

Text Generation Transformers PyTorch JAX TensorBoard

Thai gpt2 gpt2-base-thai Inference Endpoints text-generation-inference

Model card Files Files and versions Metrics Training metrics Community

gpt2-base-thai / flax_to_torch.py

w11wo's picture

pytorch model

05f5321 almost 3 years ago

raw history blame contribute delete

No virus

229 Bytes

	from transformers import GPT2LMHeadModel, AutoTokenizer

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

	tokenizer = AutoTokenizer.from_pretrained("./")
	tokenizer.save_pretrained("./")