PR #20346 [Switch Transformers] Fix failing slow test

[Switch Transformers] Fix failing slow test #20346

younesbelkada merged 3 commits into huggingface:main from younesbelkada:fix-switch-slow

run slow test on GPU

cf1bcc74

younesbelkada marked this pull request as ready for review 3 years ago

remove unnecessary device assignment

4d38e612

younesbelkada changed the title ~~run slow test on GPU~~ [Switch Transformers] Fix failing slow test 3 years ago

sgugger commented on 2022-11-21

use `torch_device` instead

b2cd9531

younesbelkada requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-11-21

younesbelkada merged 74297d0a into main 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone