Fix MoE routers returning probabilities instead of logits #45131
yacinemebarki
wants to merge
3 commits into
huggingface:main
from
yacinemebarki:fix-moe-router-logits
Fix MoE routers returning probabilities instead of logits
f9f87b7c
vasqu
approved these changes
on 2026-03-31
Merge branch 'huggingface:main' into fix-moe-router-logits
76d79984
Fix #45120: Keep router_logits raw, add router_probs separately
8c85a85d
Login to write a write a comment.
Login via GitHub