如何加载本地模型
yfangZhang opened this issue · 1 comments
yfangZhang commented
step 2
from openprompt.plms import load_plm
plm, tokenizer, model_config, WrapperClass = load_plm("bert", "bert-base-cased")
如何修改代码,加载本地模型
peterpaniff commented
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"即可。