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