riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 10:41:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity
d21c97cc0f
transformers/tests/models/switch_transformers
History
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