Skip to content

Keep float32 MoE router logits in float32 under autocast - #50003

Draft
FeSens wants to merge 1 commit into
huggingface:mainfrom
FeSens:fix/router-autocast
Draft

FeSens wants to merge 1 commit into
huggingface:mainfrom
FeSens:fix/router-autocast

Keep float32 MoE router logits in float32 under autocast

e5555d3
Select commit
Loading
Failed to load commit list.
Sign in for the full log view