thunlp / OpenPrompt

An Open-Source Framework for Prompt-Learning.
https://thunlp.github.io/OpenPrompt/
Apache License 2.0
4.32k stars 447 forks source link

如何加载本地模型 #298

Open yfangZhang opened 11 months ago

yfangZhang commented 11 months ago

step 2

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

peterpaniff commented 9 months ago

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"即可。