Merge pull request #168 from bratao/main

Disable Wandb if no wandb project is specified
axolotl-ai-cloud · Jun 9, 2023 · febe902 · febe902
2 parents 281dc3d + f4df266
commit febe902
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/axolotl/utils/wandb.py b/src/axolotl/utils/wandb.py
@@ -15,3 +15,5 @@ def setup_wandb_env_vars(cfg):
             os.environ["WANDB_LOG_MODEL"] = cfg.wandb_log_model
         if cfg.wandb_run_id and len(cfg.wandb_run_id) > 0:
             os.environ["WANDB_RUN_ID"] = cfg.wandb_run_id
+    else:
+        os.environ["WANDB_DISABLED"] = "true"