初始化项目,由ModelHub XC社区提供模型
Model: dkleczek/papuGaPT2 Source: Original Platform
This commit is contained in:
5
convert_to_pytorch.py
Executable file
5
convert_to_pytorch.py
Executable file
@@ -0,0 +1,5 @@
|
||||
#!/usr/bin/env python3
|
||||
from transformers import GPT2LMHeadModel
|
||||
|
||||
model = GPT2LMHeadModel.from_pretrained("./", from_flax=True)
|
||||
model.save_pretrained("./")
|
||||
Reference in New Issue
Block a user