程序在每次运行的时候都会链接huggingface.co去下载模型文件,比较麻烦,我们直接让它在本地电脑读取。
首先先将所有需要的模型文件下载下来,放到当前程序运行目录内如huggingface文件夹
修改HUGGINGFACE的缓存位置环境变量,Python执行脚本内修改代码:
os.environ["HUGGINGFACE_HUB_CACHE"] = os.path.join(os.getcwd(), "huggingface")
你也可以使用其它方式设置HUGGINGFACE_HUB_CACHE的系统环境变量值
然后再让HUGGINGFACE离线,使用如下方式:
set HF_DATASETS_OFFLINE=1
set TRANSFORMERS_OFFLINE=1
通过设置环境变量值OFFLINE=1来保持离线,这样别人在拿到你的程序之后运行起来就不再需要去联网检查并下载模型文件了。