flax-community
/

papuGaPT2

Text Generation

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

papuGaPT2 / convert_to_pytorch.py

dkleczek's picture

praying now

7848bdf almost 3 years ago

raw history blame contribute delete

No virus

155 Bytes

	#!/usr/bin/env python3
	from transformers import GPT2LMHeadModel

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