項目場景:
有一些模型需要在huggingface下載,因為國內限制,一般無法訪問huggingface網站進行下載。然而,可以通過國內的鏡像下載。網上大部分都是在linux系統下,通過huggingface提供的指令下載。本文針對采用python腳本在win10系統下載。
解決方案:
- 配置好python環境,下載huggingface_hub第三方庫。
- 通過cmd設置win10環境變量,該變量為臨時變量。
set HF_ENDPOINT=https://hf-mirror.com
通過以下指令查看是否設置成功
set HF_ENDPOINT
- 編寫下載腳本 downlod.py
from huggingface_hub import snapshot_download# 指定模型名稱(如 "bert-base-uncased")
model_name = "SWHL/RapidOCR"
# 指定保存路徑(可選,默認在用戶緩存目錄)
local_dir = "D:/Huggingface" #修改這里即可# 下載模型
snapshot_download(repo_id=model_name,local_dir=local_dir,local_dir_use_symlinks=False, # 禁用符號鏈接(避免權限問題)resume_download=True # 支持斷點續傳
)print(f"模型已下載到:{local_dir}")
通過運行python download.py
切記要在配置有huggingface_hub環境下運行