Younes Belkada 74297d0a55 [Switch Transformers] Fix failing slow test (#20346) ...

* run slow test on GPU

* remove unnecessary device assignment

* use `torch_device` instead

2022-11-21 15:36:49 +01:00

..

__init__.py

Add Switch transformers (#19323)

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

test_modeling_switch_transformers.py

[Switch Transformers] Fix failing slow test (#20346)

2022-11-21 15:36:49 +01:00