fix: suspected eval vram increased usage

2025-06-23 18:44:03 +07:00
1 changed files with 0 additions and 4 deletions
--- 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