自学内容网 自学内容网

部署搭建AI相关项目时,不用魔法也能轻松自动下载所需大模型

背景

最近搭建了许多AI相关的自动化服务,有些时候因为国内服务器墙了 huggingface.co 访问,导致一些依赖文件和模型下载不下来,手动去下载又特别麻烦,今天教你一个小招,轻松解决这个问题

开搞

1:首先确定下你环境里是否安装 huggingface_hub,没安装的话执行安装

pip install -U huggingface_hub

2:进入conda 虚拟环境目录,进入 huggingface_hub 包目录,例如我的安装在:

/root/conda/envs/comfyui/lib/python3.10/site-packages/huggingface_hub

3:更改包目录下的 init.py 文件
在这里插入图片描述
4:在文件最后一行加入:os.environ[“HF_ENDPOINT”] = ‘https://hf-mirror.com’
在这里插入图片描述
5:保存退出,并执行 conda init,然后重新进入虚拟环境

conda init 
conda deactivate 
conda activate xxxx

6:重启你的项目服务,就可以愉快的下载玩耍了


原文地址:https://blog.csdn.net/ltbweber/article/details/143740074

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!