初始化项目,由ModelHub XC社区提供模型
Model: imthanhlv/vigpt2medium Source: Original Platform
This commit is contained in:
5
flax_to_pt.py
Normal file
5
flax_to_pt.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from transformers import AutoModelForCausalLM
|
||||
|
||||
|
||||
pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True)
|
||||
pt_model.save_pretrained(".")
|
||||
Reference in New Issue
Block a user