diff --git a/src/axolotl/utils/wandb.py b/src/axolotl/utils/wandb.py index 90e9c2f73..d22b932cb 100644 --- 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"