THUDM / P-tuning-v2

An optimized deep prompt tuning strategy comparable to fine-tuning across scales and tasks
Apache License 2.0
1.96k stars 198 forks source link

如何加载本地模型 #63

Open px2020zjut opened 1 year ago

px2020zjut commented 1 year ago

操作环境:WIN10+VSCODE 我已经把模型下载到本地文件夹preweight,然后采用如下函数测试,系统报错 config = AutoConfig.from_pretrained(pretrained_model_name_or_path='./preweight/',trust_remote_code=True)

Exception has occurred: FileNotFoundError [Errno 2] No such file or directory: "C:\Users\Ad'min\AppData\Local\Temp\tmpwfgqi39a/./preweight/configuration_chatglm.py" File "F:\ChatGLM-6B-main\ptuning\transbert.py", line 12, in config = AutoConfig.from_pretrained(pretrained_model_name_or_path='./preweight/',trust_remote_code=True) # E.g. config (or model) was saved using save_pretrained('./test/saved_model/') FileNotFoundError: [Errno 2] No such file or directory: "C:\Users\Ad'min\AppData\Local\Temp\tmpwfgqi39a/./preweight/configuration_chatglm.py"