Upload folder using huggingface_hub

hf_training/train.py

@@ -17,13 +17,26 @@ from huggingface_hub import login, HfApi, hf_hub_download, snapshot_download
 from peft import set_peft_model_state_dict
 
 # ─── Auth & Config ────────────────────────────────────────────────────────────
-HF_TOKEN    = os.environ["HF_TOKEN"]
+HF_TOKEN = os.environ.get("HF_TOKEN") or os.environ.get("HUGGING_FACE_HUB_TOKEN")
+
+if not HF_TOKEN:
+    print("\n" + "="*80)
+    print(" MISSING HF_TOKEN SECRET ".center(80, "#"))
+    print(" 1. Go to your Space 'Settings' tab. ".center(80))
+    print(" 2. Find 'Variables and Secrets' section. ".center(80))
+    print(" 3. Click 'New Secret', name it 'HF_TOKEN'. ".center(80))
+    print(" 4. Paste your WRITE-access token (huggingface.co/settings/tokens). ".center(80))
+    print(" 5. Restart the Space. ".center(80))
+    print("".center(80, "#"))
+    print("="*80 + "\n")
+    sys.exit(0)  # Exit gracefully
+
 HF_USERNAME = os.environ.get("HF_USERNAME", "YashashMathur")
 STEP50_REPO = f"{HF_USERNAME}/aegis-step50"
 CKPT_REPO   = f"{HF_USERNAME}/aegis-training-checkpoints"
 
 login(token=HF_TOKEN)
-api = HfApi()
+api = HfApi(token=HF_TOKEN)
 
 # Optional WandB Logging
 WANDB_API_KEY = os.environ.get("WANDB_API_KEY")