bot commited on
Commit
94ecc19
·
1 Parent(s): 44038e9

Skip wandb artifact for large models (uploaded to HF instead)

Browse files
lerobot/src/lerobot/rl/wandb_utils.py CHANGED
@@ -145,7 +145,10 @@ class WandBLogger:
145
  if config_file.exists():
146
  artifact.add_file(config_file)
147
  elif standard_model_file.exists():
148
- # Standard model: add the single safetensors file
 
 
 
149
  artifact.add_file(standard_model_file)
150
  else:
151
  logging.warning(
 
145
  if config_file.exists():
146
  artifact.add_file(config_file)
147
  elif standard_model_file.exists():
148
+ # Skip large model files (>1GB) - checkpoints are uploaded to HF instead
149
+ if standard_model_file.stat().st_size > 1e9:
150
+ logging.info(f"Skipping WandB artifact for large model ({standard_model_file.stat().st_size / 1e9:.1f}GB)")
151
+ return
152
  artifact.add_file(standard_model_file)
153
  else:
154
  logging.warning(