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)