Yih-Dar 82dd6c14bb 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>

2025-01-09 15:36:22 +01:00

..

__init__.py

Add Switch transformers (#19323)

2022-11-15 13:06:45 +01:00

test_modeling_switch_transformers.py

Fix flaky SwitchTransformersModelTest::test_training_gradient (#35587)

2025-01-09 15:36:22 +01:00