OpenPrompt 如何加载本地模型

kiz8lqtg  于 6个月前  发布在  其他
关注(0)|答案(1)|浏览(107)

step 2

from openprompt.plms import load_plm
plm, tokenizer, model_config, WrapperClass = load_plm("bert", "bert-base-cased")
如何修改代码,加载本地模型

hwamh0ep

hwamh0ep1#

step 2

from openprompt.plms import load_plm plm, tokenizer, model_config, WrapperClass = load_plm("bert", "bert-base-cased") 如何修改代码,加载本地模型
huggingface 上搜索关键字bert-base-cased,进入 https://huggingface.co/bert-base-cased/tree/main,将文件保存下来保存到某个目录,比如你保存到/home/123/bert_model/,这个路径根据你的需要进行修改,然后将这个路径替换掉"bert-base-cased"即可。

相关问题