remove use_vessl variable

Y-IAB · Mar 29, 2024 · 74511a4 · 74511a4
1 parent 0b76c97
commit 74511a4
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
@@ -854,7 +854,7 @@ def get_callbacks(self) -> List[TrainerCallback]:
                 SaveAxolotlConfigtoWandBCallback(self.cfg.axolotl_config_path)
             )
 
-        if self.cfg.use_vessl:
+        if self.cfg.vessl_credential_path:
             from axolotl.utils.callbacks.vessl_ import VesslLogMetricsCallback
 
             callbacks.append(

diff --git a/src/axolotl/utils/vessl_.py b/src/axolotl/utils/vessl_.py
@@ -10,5 +10,4 @@ def setup_vessl_env_vars(cfg: DictDefault):
     # default credential inside a VESSL Run
     credential_path = os.environ.get("VESSL_RUN_INITIAL_CONFIG")
     if credential_path:
-        cfg.use_vessl = True
         cfg.vessl_credential_path = credential_path