From 91dae42737411dbbdcf43b90a0dc82e6125a5481 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Wed, 25 Feb 2026 15:04:28 +0700 Subject: [PATCH] fix: re-add patch from transformers PR #39866 --- src/axolotl/core/trainers/base.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/axolotl/core/trainers/base.py b/src/axolotl/core/trainers/base.py index d055608dd..76e8f105f 100644 --- a/src/axolotl/core/trainers/base.py +++ b/src/axolotl/core/trainers/base.py @@ -765,7 +765,11 @@ class AxolotlTrainer( metadata={"format": "pt"}, ) else: - self.model.save_pretrained(output_dir, state_dict=state_dict) + self.model.save_pretrained( + output_dir, + state_dict=state_dict, + is_main_process=self.accelerator.is_main_process, + ) if self.processing_class is not None: self.processing_class.save_pretrained(output_dir)