Skip to content

Commit a68a716

Browse files
committed
remove default writer
1 parent 7898fa6 commit a68a716

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mftcoder_accelerate/src/mpt/mpt_trainer.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ def accelerate_saving_states(self, output_dir: str, completed_steps: int):
189189
logger.info(f"[CHECKPOINT] Saving checkpoint states")
190190
self.accelerator.save_state(output_dir)
191191
self.accelerator.wait_for_everyone()
192-
192+
193193
# save safetensors for direct inference if needed
194194
if self.args.save_transformers_model:
195195
logger.info(f"[CHECKPOINT] Saving transformers(hf) model", main_process_only=True)
@@ -202,7 +202,7 @@ def accelerate_saving_states(self, output_dir: str, completed_steps: int):
202202
state_dict=self.accelerator.get_state_dict(self.model),
203203
)
204204
self.accelerator.wait_for_everyone()
205-
205+
206206
# tokenizer saving and bug dummy ckpt cleaning.
207207
if self.accelerator.is_main_process:
208208
if self.args.model_type.lower() == "deepseek":
@@ -211,7 +211,7 @@ def accelerate_saving_states(self, output_dir: str, completed_steps: int):
211211
)
212212
else:
213213
self.tokenizer.save_pretrained(output_dir)
214-
214+
215215
sf = os.path.join(output_dir, "model.safetensors")
216216
index_file = os.path.join(output_dir, "model.safetensors.index.json")
217217
if os.path.isfile(sf) and os.path.isfile(index_file):

0 commit comments

Comments
 (0)