DeepSpeed
3bc882fa - Fix Muon optimizer module path (#7802)

Commit
12 days ago
Fix Muon optimizer module path (#7802) Update the Muon optimizer import to its current location under deepspeed.runtime.zero.muon, so it’s correctly registered as a supported ZeRO optimizer. This prevents the “untested optimizer” assertion from firing. Signed-off-by: Masahiro Tanaka <mtanaka@anyscale.com>
Author
Parents
Loading