tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: re-add patch from transformers PR #39866

Browse Source
This commit is contained in:
NanoCode012
2026-02-25 15:04:28 +07:00
parent 4657cb7177
commit 91dae42737
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/axolotl/core/trainers/base.py
View File

@@ -765,7 +765,11 @@ class AxolotlTrainer(
metadata={"format": "pt"}, metadata={"format": "pt"},
) )
else: 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: if self.processing_class is not None:
self.processing_class.save_pretrained(output_dir) self.processing_class.save_pretrained(output_dir)