From d260eeb57df1848ccc0b8cf3ab268b9dcd0e321b Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 15 Feb 2026 07:55:55 -0500 Subject: [PATCH] match protected method --- src/axolotl/core/trainers/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)