DeepSpeed
MOE: Fix save checkpoint when TP > 1
#5157
Merged

MOE: Fix save checkpoint when TP > 1 #5157

tjruwase merged 1 commit into deepspeedai:master from mosheisland:moe/ckp
mosheisland
mosheisland mosheisland requested a review from mrwyattii mrwyattii 1 year ago
mosheisland mosheisland requested a review from tjruwase tjruwase 1 year ago
tjruwase
tjruwase approved these changes on 2024-02-19
MOE: Fix save checkpoint when TP > 1
64c099ab
mosheisland mosheisland force pushed from bbaa158f to 64c099ab 1 year ago
tjruwase tjruwase merged a84d07c5 into master 1 year ago
mosheisland mosheisland deleted the moe/ckp branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone