match protected method

2026-02-15 07:55:55 -05:00
parent 5a7f007d20
commit d260eeb57d
1 changed files with 1 additions and 1 deletions
--- 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)