Upload train_sft_hf_cloud.py with huggingface_hub
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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!")
|