File size: 154 Bytes
be2735b |
1 2 3 4 5 6 |
#!/usr/bin/env pyton3
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")
|
be2735b |
1 2 3 4 5 6 |
#!/usr/bin/env pyton3
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")
|