Upload 2025-03-09/test.sh with huggingface_hub
Browse files- 2025-03-09/test.sh +4 -2
2025-03-09/test.sh
CHANGED
@@ -2,14 +2,16 @@ echo "Hello Chenghao!"
|
|
2 |
|
3 |
# 设置 Hugging Face 环境
|
4 |
setup_huggingface() {
|
5 |
-
pip install -U "huggingface_hub[cli]"
|
|
|
|
|
6 |
# 确保 HF_TOKEN 已经设置为环境变量
|
7 |
if [ -z "$HF_TOKEN" ]; then
|
8 |
echo "错误:HF_TOKEN 环境变量未设置。请在运行此脚本之前设置它。"
|
9 |
return 1
|
10 |
fi
|
11 |
|
12 |
-
huggingface-cli login --token "$HF_TOKEN" --add-to-git-credential
|
13 |
|
14 |
# 验证是否包含 megatrump
|
15 |
WHOAMI_OUTPUT=$(huggingface-cli whoami)
|
|
|
2 |
|
3 |
# 设置 Hugging Face 环境
|
4 |
setup_huggingface() {
|
5 |
+
pip install -U "huggingface_hub[cli]" > /dev/null 2>&1
|
6 |
+
pip install -U "huggingface_hub[hf_transfer]" > /dev/null 2>&1
|
7 |
+
export HF_HUB_ENABLE_HF_TRANSFER=1
|
8 |
# 确保 HF_TOKEN 已经设置为环境变量
|
9 |
if [ -z "$HF_TOKEN" ]; then
|
10 |
echo "错误:HF_TOKEN 环境变量未设置。请在运行此脚本之前设置它。"
|
11 |
return 1
|
12 |
fi
|
13 |
|
14 |
+
huggingface-cli login --token "$HF_TOKEN" --add-to-git-credential > /dev/null 2>&1
|
15 |
|
16 |
# 验证是否包含 megatrump
|
17 |
WHOAMI_OUTPUT=$(huggingface-cli whoami)
|