mlopez6132 commited on
Commit
695e015
·
verified ·
1 Parent(s): ee850c7

Upload hf_free_training.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- login() # Will use HF_TOKEN environment variable
92
- api = HfApi()
 
 
 
 
 
 
 
 
 
 
 
 
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