Skip to content

[Fix] Fix device mismatch for ascend npu when loss_cfg.loss_reduction != "token"#1365

Open
nil0x9 wants to merge 1 commit intoInternLM:mainfrom
nil0x9:linty/fix-npu-loss-weight-device-mismatch
Open

[Fix] Fix device mismatch for ascend npu when loss_cfg.loss_reduction != "token"#1365
nil0x9 wants to merge 1 commit intoInternLM:mainfrom
nil0x9:linty/fix-npu-loss-weight-device-mismatch

Commits