File size: 210 Bytes
aa4d81d |
1 2 3 4 5 6 7 8 9 |
from transformers import GPT2Config, GPT2LMHeadModel
'''
This is a script to convert the Jax model to Pytorch model
'''
model = GPT2LMHeadModel.from_pretrained(".", from_flax=True)
model.save_pretrained(".")
|