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)
|