transformers
[Switch Transformers] Fix failing slow test
#20346
Merged

[Switch Transformers] Fix failing slow test #20346

younesbelkada
younesbelkada run slow test on GPU
cf1bcc74
younesbelkada younesbelkada marked this pull request as ready for review 3 years ago
HuggingFaceDocBuilderDev
younesbelkada remove unnecessary device assignment
4d38e612
younesbelkada younesbelkada changed the title run slow test on GPU [Switch Transformers] Fix failing slow test 3 years ago
sgugger
sgugger commented on 2022-11-21
younesbelkada use `torch_device` instead
b2cd9531
younesbelkada
younesbelkada younesbelkada requested a review from sgugger sgugger 3 years ago
sgugger
sgugger approved these changes on 2022-11-21
younesbelkada younesbelkada merged 74297d0a into main 3 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone