【AI模型部署】

解决python引入huggingface_hub模块下载超时问题

背景

AMD Ryzen™ AI处理器通过独特的NPU+GPU异构架构,为AI工作负载提供强大的并行计算能力。本方案展示了如何将YOLOv8目标检测、RCAN超分辨率重建和Stable Diffusion文生图三类模型分别部署到NPU和GPU,实现计算资源的智能调度。

问题

本地部署模型时,py代码中引入huggingface_hub模块并执行时,报下载超时。

from huggingface_hub import model_info
raise ConnectTimeout(e, request=request)
requests.exceptions.ConnectTimeout: (MaxRetryError("HTTPSConnectionPool(host='huggingface.co', port=443): Max retries exceeded with url: /api/models/runwayml/stable-diffusion-v1-5 (Caused by ConnectTimeoutError(<urllib3.connection.HTTPSConnection object at 0x00000222DC116050>, 'Connection to huggingface.co timed out. (connect timeout=None)'))"), '(Request ID: d0ef776a-1cf6-4c07-a260-908d1684f2ae)')

解决

  1. 在当前python虚拟环境终端下执行:
pip show huggingface_hub | findstr Location
  1. 找到huggingface_hub包,并打开:
    在这里插入图片描述
  2. 找到constants.py文件,并修改:将huggingface.co全部替换为hf-mirror.com即可。
    在这里插入图片描述
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值