transformers

82dd6c14 - Fix flaky `SwitchTransformersModelTest::test_training_gradient` (#35587)

Fix flaky `SwitchTransformersModelTest::test_training_gradient` (#35587) * fix * Update tests/models/switch_transformers/test_modeling_switch_transformers.py Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com> --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com> Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>