accelerate
[tests] use `torch_device` instead of `0` for device check
#2861
Merged

[tests] use `torch_device` instead of `0` for device check #2861

muellerzr merged 4 commits into huggingface:main from faaany:device-idx
faaany
faaany bug fix
06e33f81
faaany fix one more case
ec60c067
faaany add more cases
29ecd2ac
SunMarc
SunMarc approved these changes on 2024-06-17
SunMarc SunMarc requested a review from muellerzr muellerzr 2 years ago
HuggingFaceDocBuilderDev
faaany refine
349154c6
faaany
muellerzr
muellerzr approved these changes on 2024-06-18
muellerzr muellerzr merged 5d4a3beb into main 2 years ago
faaany faaany deleted the device-idx branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone