fix: re-add patch from transformers PR #39866

2026-02-25 15:04:28 +07:00
parent 4657cb7177
commit 91dae42737
1 changed files with 5 additions and 1 deletions
--- 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)