Skip to content

Commit

Permalink
engine.py: remove unused _curr_save_path (#4844)
Browse files Browse the repository at this point in the history
Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
  • Loading branch information
nelyahu and mrwyattii authored Dec 20, 2023
1 parent 9e455d7 commit 1864391
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion deepspeed/runtime/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -3212,7 +3212,6 @@ def _save_moe_checkpoint(self, save_dir, tag, client_state={}, exclude_frozen_pa
state.update(client_state)
logger.info(f'Saving model checkpoint: {save_path}')
self.checkpoint_engine.save(state, save_path)
self._curr_save_path = None

def _create_checkpoint_file(self, save_dir, tag, zero_checkpoint):
name_function = (self._get_zero_ckpt_name if zero_checkpoint else self._get_ckpt_name)
Expand Down

0 comments on commit 1864391

Please sign in to comment.