bark因网络问题从huggingface预加载到本地使用时报错

最新推荐文章于 2024-09-27 10:58:53 发布

Granday

最新推荐文章于 2024-09-27 10:58:53 发布

阅读量206

点赞数 2

文章标签：人工智能 python

本文链接：https://blog.csdn.net/granday/article/details/140493742

版权

因为服务器的网络原因，无法从huggingface上直接加载模型，需要预先将模型下载到本地，指定本地路径进行from_pretrained使用。

但是在执行下述代码时还是报了从hub上下载出现的网络错误：

processor = AutoProcessor.from_pretrained(model_name,trust_remote_code=True)
model = BarkModel.from_pretrained(model_name,trust_remote_code=True)

# 指定特定voice_preset时出错
inputs = self.processor(text, voice_preset=voice_preset, return_tensors="pt")

报错信息：

查看源码后发现是使用了模型路径下的`speaker_embeddings_path.json`中的`repo_or_path`字段值，该字段值为：将该字段值改为自己本地存储bark-small模型的路径即可。