transformers/tests/models/switch_transformers
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