一、Huggingface镜像站的使用方法
使用Huggingface镜像站非常简单,只需要按照以下步骤进行设置即可:
安装依赖:首先,确保你的计算机上已经安装了pip这个Python包管理工具。然后,通过以下命令安装huggingface_hub和hf_transfer这两个库:
pip install -U huggingface_hub hf_transfer
如果你想使用清华源加速下载,可以加上-i https://pypi.tuna.tsinghua.edu.cn/simple参数:
pip install -U huggingface_hub hf_transfer -i https://pypi.tuna.tsinghua.edu.cn/simple
设置环境变量:接下来,设置环境变量HF_ENDPOINT为Huggingface镜像站的地址:
export HF_ENDPOINT=https://hf-mirror.com
这样,后续的命令就会使用这个镜像站点而非Huggingface的主仓库。
下载模型:现在,你可以使用huggingface-cli命令来下载模型了。例如,要下载名为“openai-community/gpt2-medium”的模型,可以运行以下命令:
huggingface-cli download --resume-download openai-community/gpt2-medium --local-dir /path/to/local/directory
其中,--local-dir参数指定了模型下载的本地目录。你可以将其替换为你想要保存模型的路径。