megatrump commited on
Commit
1a8eb0c
·
verified ·
1 Parent(s): 996edf4

Upload 2025-03-09/test.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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)