yangluo commited on
Commit
9386de3
·
verified ·
1 Parent(s): 9d01f39

Upload train_sft_hf_cloud.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_sft_hf_cloud.py +6 -1
train_sft_hf_cloud.py CHANGED
@@ -104,7 +104,12 @@ def main():
104
  hf_token = os.environ.get("HF_TOKEN")
105
  if hf_token:
106
  print("Logging in to Hugging Face Hub...")
107
- login(token=hf_token)
 
 
 
 
 
108
  else:
109
  print("WARNING: HF_TOKEN not found in environment!")
110
  print("Model will NOT be pushed to Hub!")
 
104
  hf_token = os.environ.get("HF_TOKEN")
105
  if hf_token:
106
  print("Logging in to Hugging Face Hub...")
107
+ try:
108
+ login(token=hf_token, add_to_git_credential=False)
109
+ print(" Login successful!")
110
+ except Exception as e:
111
+ print(f" Login failed: {e}")
112
+ print(" Continuing without explicit login (will use default credentials)...")
113
  else:
114
  print("WARNING: HF_TOKEN not found in environment!")
115
  print("Model will NOT be pushed to Hub!")