papuGaPT2 / convert_to_pytorch.py
dkleczek's picture
Copying papuGaPT2 Flax model repo
3b456c2
#!/usr/bin/env python3
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")