Skip to content
1 change: 1 addition & 0 deletions deepspeed/runtime/engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -3601,6 +3601,7 @@ def _save_moe_checkpoint(self, save_dir, tag, client_state={}, exclude_frozen_pa
local_expert_id = None
if not m:
logger.warning(f'No expert found in key {key}.')
continue
else:
local_expert_id = m.group(1)

Expand Down
Loading