Skip to content

Commit 9e7d3c6

Browse files
committed
if no expert found in parameter that have expert in name it should continue
Otherwise: `TypeError: int() argument must be a string, a bytes-like object or a real number, not 'NoneType'` Signed-off-by: Luke Friedrichs <[email protected]>
1 parent 91172a7 commit 9e7d3c6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

deepspeed/runtime/engine.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3601,6 +3601,7 @@ def _save_moe_checkpoint(self, save_dir, tag, client_state={}, exclude_frozen_pa
36013601
local_expert_id = None
36023602
if not m:
36033603
logger.warning(f'No expert found in key {key}.')
3604+
continue
36043605
else:
36053606
local_expert_id = m.group(1)
36063607

0 commit comments

Comments
 (0)