fix: suspected eval vram increased usage

2025-06-23 18:44:03 +07:00
2 changed files with 1 additions and 5 deletions
--- a/src/axolotl/init.py
+++ b/src/axolotl/init.py
@@ -4,4 +4,4 @@ import pkgutil

 __path__ = pkgutil.extend_path(__path__, __name__)  # Make this a namespace package

-__version__ = "0.10.1"
+__version__ = "0.11.0.dev"
--- a/src/axolotl/core/builders/causal.py
+++ b/src/axolotl/core/builders/causal.py
@@ -215,10 +215,6 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
                self.cfg.auto_find_batch_size
            )

-        training_arguments_kwargs["eval_accumulation_steps"] = (
-            self.cfg.gradient_accumulation_steps
-        )
-
        training_arguments_kwargs["load_best_model_at_end"] = (
            (
                self.cfg.load_best_model_at_end is not False