from pyannote.audio import Model
from pyannote.audio import Inference
原代码:
model = Model.from_pretrained("pyannote/embedding", use_auth_token=HF_TOKEN)
这个代码由于网络问题无法自动下载,因此去huggingface手动下载了全部文件到本地:/data/pyannote/embedding/下
修改代码指向本地下载文件:
model = Model.from_pretrained("/data/pyannote/embedding/", use_auth_token=HF_TOKEN)
失败:huggingface_hub.errors.HFValidationError: Repo id must be in the form 'repo_name' or 'namespace/repo_name':
解决,指向bin模型:
model = Model.from_pretrained("/data/pyannote/embedding/pytorch_model.bin", use_auth_token=HF_TOKEN)
思考: Model.from_pretraine本身是加载一个模型而已,指定一个权重即可,不需要制定一个文件夹路径(例如分词器等可能指定的是文件路径,此处可多尝试)
失败尝试:
1:
import os
os.environ['TRANSFORMERS_OFFLINE'] = "1"
2:配置参数指定本地:local_files_only=True
3:本地存储文件夹修改为类似模型id结构
4:注意linux的绝对路径开头必须是 /xxxxx