transformers

97266dfd - Fix flaky `JambaModelTest.test_load_balancing_loss` (#40617)

Fix flaky `JambaModelTest.test_load_balancing_loss` (#40617) * fix * fix * fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>