transformers/tests/models/switch_transformers
Joao Gante 248d5d23a2
Tests: replace `torch.testing.assert_allclose` by `torch.testing.assert_close` (#29915)
* replace torch.testing.assert_allclose by torch.testing.assert_close

* missing atol rtol
2024-03-28 09:53:31 +00:00
..
__init__.py Add Switch transformers (#19323) 2022-11-15 13:06:45 +01:00
test_modeling_switch_transformers.py Tests: replace `torch.testing.assert_allclose` by `torch.testing.assert_close` (#29915) 2024-03-28 09:53:31 +00:00