diff --git a/src/axolotl/core/trainers/base.py b/src/axolotl/core/trainers/base.py index 2a8c07a68..755fa3f8b 100644 --- a/src/axolotl/core/trainers/base.py +++ b/src/axolotl/core/trainers/base.py @@ -605,7 +605,7 @@ class AxolotlTrainer( return args - def build_accelerator_args(self, **kwargs) -> dict[str, Any]: + def _build_accelerator_args(self, **kwargs) -> dict[str, Any]: args = super().build_accelerator_args(**kwargs) fp8_args = self.build_fp8_accelerator_args() args.update(fp8_args)