papuGaPT2-large / convert_to_pytorch.py
miwojc's picture
convert to pytorch
be2735b
raw
history blame contribute delete
154 Bytes
#!/usr/bin/env pyton3
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")