dkleczek
/

papuGaPT2

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

papuGaPT2 / convert_to_pytorch.py

dkleczek's picture

Copying papuGaPT2 Flax model repo

3b456c2 about 3 years ago

155 Bytes

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

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