Visual Code的Terminal （PowerShell）中下载huggingface模型

Tonyfield2024-12-08 10:46

方案：使用 huggingface_hub 工具 + hf-mirror 镜像站点实现

1. 安装 huggingface_hub 工具

bash 复制代码

python -m pip install -U huggingface_hub -i https://pypi.tuna.tsinghua.edu.cn/simple/

2. 设置环境变量：env:HF_ENDPOINT

huggingface-cli 指令将用指定的站点代替缺省站点 huggingface.co

bash 复制代码

$env:HF_ENDPOINT = "https://hf-mirror.com"
$env:HF_INFERENCE_ENDPOINT = "https://api-inference.hf-mirror.com"

确认是否设置成功

bash 复制代码

PS K:\> huggingface-cli.exe env

Copy-and-paste the text below in your GitHub issue.

- huggingface_hub version: 0.26.5
。。。。。
- ENDPOINT: https://hf-mirror.com   <--- 修改成功了
。。。

3. 设置环境变量：$env:PATH

将 huggingface-cli.exe所在路径加入PATH

bash 复制代码

$env:PATH += "C:\Users\your-user-name\AppData\Local\Programs\Python\Python310\Scripts"

4. 下载你需要的模型，下以 layoutlmv3-base 为例

bash 复制代码

huggingface-cli.exe download microsoft/layoutlmv3-base

【可选】设置环境变量 HF_HOME，将模型保存在当前路径下

bash 复制代码

$env:HF_HOME = "."

检查是否生效

bash 复制代码

huggingface-cli env
。。。
- HF_HUB_CACHE: .\hub
- HF_ASSETS_CACHE: .\assets
- HF_TOKEN_PATH: .\token
- HF_STORED_TOKENS_PATH: .\stored_tokens