From 6a342feda227f6c0676832be38340885aa70f921 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 9 Dec 2024 19:24:48 +0700 Subject: [PATCH] fix: duplicate mlflow logging (#2109) [skip ci] --- src/axolotl/core/trainer_builder.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index 691437bc65..73d9e0e655 100755 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -1368,8 +1368,6 @@ def get_callbacks(self) -> List[TrainerCallback]: SaveAxolotlConfigtoWandBCallback(self.cfg.axolotl_config_path) ) if self.cfg.use_mlflow and is_mlflow_available(): - from transformers.integrations.integration_utils import MLflowCallback - from axolotl.utils.callbacks.mlflow_ import ( SaveAxolotlConfigtoMlflowCallback, ) @@ -1377,7 +1375,6 @@ def get_callbacks(self) -> List[TrainerCallback]: callbacks.extend( [ SaveAxolotlConfigtoMlflowCallback(self.cfg.axolotl_config_path), - MLflowCallback, ] ) if self.cfg.use_comet and is_comet_available():