transformers
2e24ee4d - Fix typing in `load_balancing_loss_func` function of `modeling_mixtral.py`. (#33641)

Commit
1 year ago
Fix typing in `load_balancing_loss_func` function of `modeling_mixtral.py`. (#33641) * fix return type * update to union * fix gate_logits typing * fix num_experts type * fix typing * run fix-copies * add doc for top_k * run fix-copies * empty commit to trigger CI
Author
Parents
Loading