Upload hf_free_training.py with huggingface_hub

hf_free_training.py

@@ -88,8 +88,20 @@ config = {k: globals()[k] for k in config_keys}
 
 # HF setup
 if push_to_hub:
-    login()  # Will use HF_TOKEN environment variable
-    api = HfApi()
+    # Check if HF_TOKEN environment variable is set
+    if os.environ.get('HF_TOKEN'):
+        login(token=os.environ.get('HF_TOKEN'))
+    else:
+        # Try to login without token (will use cached credentials)
+        try:
+            login()
+        except Exception as e:
+            print(f"Warning: Could not login to HF Hub: {e}")
+            print("Continuing without HF Hub upload...")
+            push_to_hub = False
+    
+    if push_to_hub:
+        api = HfApi()
 
 # various inits, derived attributes, I/O setup
 ddp = int(os.environ.get('RANK', -1)) != -1