tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: duplicate mlflow logging (#2109) [skip ci]

Browse Source
This commit is contained in:
NanoCode012
2024-12-09 19:24:48 +07:00
committed by GitHub
parent 0c25bc07a2
commit 6a342feda2
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/core/trainer_builder.py
View File

@@ -1368,8 +1368,6 @@ class TrainerBuilderBase(abc.ABC):
SaveAxolotlConfigtoWandBCallback(self.cfg.axolotl_config_path) SaveAxolotlConfigtoWandBCallback(self.cfg.axolotl_config_path)
) )
if self.cfg.use_mlflow and is_mlflow_available(): if self.cfg.use_mlflow and is_mlflow_available():
from transformers.integrations.integration_utils import MLflowCallback
from axolotl.utils.callbacks.mlflow_ import ( from axolotl.utils.callbacks.mlflow_ import (
SaveAxolotlConfigtoMlflowCallback, SaveAxolotlConfigtoMlflowCallback,
) )
@@ -1377,7 +1375,6 @@ class TrainerBuilderBase(abc.ABC):
callbacks.extend( callbacks.extend(
[ [
SaveAxolotlConfigtoMlflowCallback(self.cfg.axolotl_config_path), SaveAxolotlConfigtoMlflowCallback(self.cfg.axolotl_config_path),
MLflowCallback,
] ]
) )
if self.cfg.use_comet and is_comet_available(): if self.cfg.use_comet and is_comet_available():