Spaces:
Sleeping
Sleeping
Upload hf_free_training.py with huggingface_hub
Browse files- hf_free_training.py +14 -2
hf_free_training.py
CHANGED
@@ -88,8 +88,20 @@ config = {k: globals()[k] for k in config_keys}
|
|
88 |
|
89 |
# HF setup
|
90 |
if push_to_hub:
|
91 |
-
|
92 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
93 |
|
94 |
# various inits, derived attributes, I/O setup
|
95 |
ddp = int(os.environ.get('RANK', -1)) != -1
|
|
|
88 |
|
89 |
# HF setup
|
90 |
if push_to_hub:
|
91 |
+
# Check if HF_TOKEN environment variable is set
|
92 |
+
if os.environ.get('HF_TOKEN'):
|
93 |
+
login(token=os.environ.get('HF_TOKEN'))
|
94 |
+
else:
|
95 |
+
# Try to login without token (will use cached credentials)
|
96 |
+
try:
|
97 |
+
login()
|
98 |
+
except Exception as e:
|
99 |
+
print(f"Warning: Could not login to HF Hub: {e}")
|
100 |
+
print("Continuing without HF Hub upload...")
|
101 |
+
push_to_hub = False
|
102 |
+
|
103 |
+
if push_to_hub:
|
104 |
+
api = HfApi()
|
105 |
|
106 |
# various inits, derived attributes, I/O setup
|
107 |
ddp = int(os.environ.get('RANK', -1)) != -1
|