初始化项目,由ModelHub XC社区提供模型
Model: imthanhlv/vigpt2medium Source: Original Platform
This commit is contained in:
4
create_config.py
Normal file
4
create_config.py
Normal file
@@ -0,0 +1,4 @@
|
||||
from transformers import GPT2Config
|
||||
|
||||
config = GPT2Config.from_pretrained("gpt2-medium", resid_pdrop=0.0, embd_pdrop=0.0, attn_pdrop=0.0, vocab_size=50257)
|
||||
config.save_pretrained(".")
|
||||
Reference in New Issue
Block a user