papuGaPT2 / convert_to_pytorch.py
dkleczek's picture
praying now
7848bdf
raw
history blame
155 Bytes
#!/usr/bin/env python3
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")