transformers

Fix MoE routers returning probabilities instead of logits

#45131

Open