我已经用自定义语料库对bert模型进行了预训练,然后得到了vocab文件,检查点,model.bin,tfrecords等。
然后我按如下方式加载模型:
# Load pre-trained model (weights)
model = BertModel.from_pretrained('/content/drive/My Drive/Anirban_test_pytorch')
但是,当我尝试将模型用于任何任务(例如q和a,掩码字的预测等)时,错误下方的错误提示
from transformers import pipeline
nlp = pipeline("fill-mask", model="model")
nlp(f"This is the best thing I've {nlp.tokenizer.mask_token} in my life.")
错误:
OSError:无法加载“模型”的配置。确保:
“模型”是在'https://huggingface.co/models'
上列出的正确模型标识符或'model'是包含config.json文件的目录的正确路径
你能帮我吗?