gpt2-swahili / flax_to_torch.py
alokmatta
gpt2 swahili
49a1166
raw
history blame
131 Bytes
from transformers import GPT2LMHeadModel
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
model.save_pretrained("./")