我用google colab来尝试这个,我用snapshot_download从huggingface hub下载了《变形金刚》的模型。对于这个模型,我用的是all-MiniLM-L 6-v2
当我在同一个运行时从我下载到驱动器的repo中加载模型时,它工作得很好。
但是当我重新启动运行时,并使用之前使用的相同目录再次加载模型时,出现错误。错误是“Error no file named pytorch_model.bin,tf_model.h5,model.ckpt.index or flax_model.msgpack found in directory /content/drive/MyDrive/Generative_AI/NLP/all-MiniLM-L 6-v2/”。
怎么办?
#This is how I download the model
from huggingface_hub import snapshot_download
snapshot_download(repo_id="sentence-transformers/all-MiniLM-L6-v2",
local_dir="/content/drive/MyDrive/Generative_AI/NLP/all-MiniLM-L6-v2")
#This how I load the model
from sentence_transformers import SentenceTransformer
model_dir = '/content/drive/MyDrive/Generative_AI/NLP/all-MiniLM-L6-v2'
model = SentenceTransformer(model_dir)
字符串
我尝试重新启动运行时,仍然出现错误。原来我需要在同一个运行时下载和加载模型,这意味着我必须下载每个新的运行时。但我想下载模型一次
1条答案
按热度按时间ktca8awb1#
这个问题主要是由变压器版本引起的。我建议安装变压器==4.32.0或更高版本。
字符串