6 lines
154 B
Python
6 lines
154 B
Python
|
#!/usr/bin/env pyton3
|
||
|
|
from transformers import GPT2LMHeadModel
|
||
|
|
|
||
|
|
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
|
||
|
|
model.save_pretrained("./")
|