赞
踩
要设置HuggingFace Hub的国内镜像,你可以按照以下步骤进行操作,以确保从国内镜像站点下载Hugging Face模型和数据集。
首先,确保你已经安装了huggingface_hub
库。如果没有,可以通过pip进行安装:
bash复制代码
pip install -U huggingface_hub |
对于Linux或macOS系统,你可以在.bashrc
、.bash_profile
、.zshrc
等shell配置文件中添加以下行来永久设置环境变量:
bash复制代码
export HF_ENDPOINT=https://hf-mirror.com |
然后,你需要使这些更改生效:
bash复制代码
source ~/.bashrc # 或者 source ~/.bash_profile, source ~/.zshrc 等 |
对于Windows系统,你可以通过“系统属性”中的“环境变量”来设置HF_ENDPOINT
环境变量。
如果你只想在当前终端会话中设置环境变量,可以直接在命令行中输入:
对于Linux或macOS:
bash复制代码
export HF_ENDPOINT=https://hf-mirror.com |
对于Windows(PowerShell):
powershell复制代码
$env:HF_ENDPOINT = "https://hf-mirror.com" |
一旦设置了环境变量,你就可以使用Hugging Face CLI工具来下载模型和数据集了。例如:
bash复制代码
huggingface-cli download --resume-download --local-dir gpt2 gpt2 |
这条命令将会从国内镜像站点下载GPT-2模型,并将其保存在本地的gpt2
目录中。
对于某些受限访问的模型,你可能需要提供Hugging Face账户的访问令牌(Token)才能下载。你可以在Hugging Face官网的个人设置中找到这个Token,并将其添加到你的Python脚本或环境变量中。但是,请注意,这通常不是必需的,除非你正在尝试下载受限的模型。
huggingface_hub
库并设置环境变量。Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。