Home

gqjia

29 Nov 2023

huggingface 下载模型和数据集

step1: 安装 huggingface_hub 和 hf-transfer

pip install -U huggingface_hub
pip install -U hf-transfer

hf-transfer 及后续环境配置是可选的。

如果需要清华镜像源:

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

step2: 设置环境变量

Linux

export HF_ENDPOINT=https://hf-mirror.com 
export HF_HUB_ENABLE_HF_TRANSFER=1

Windows Powershell

$env:HF_ENDPOINT = "https://hf-mirror.com"
$env:HF_HUB_ENABLE_HF_TRANSFER = 1

python

import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

step3: 下载

huggingface-cli download --resume-download bigscience/bloom-560m --local-dir bloom-560m  --local-dir-use-symlinks False

如果需要 huggingface token 需要加上 --token hf_xxxx

Til next time,
gqjia at 00:00

scribble