Skip to content

fix(testing_utils): guard get_device_capability() with torch.cuda.is_available() - #45427

Closed
Aftabbs wants to merge 1 commit into
huggingface:mainfrom
Aftabbs:fix/testing-utils-cuda-available-check
Closed

Aftabbs wants to merge 1 commit into
huggingface:mainfrom
Aftabbs:fix/testing-utils-cuda-available-check

fix(testing_utils): guard get_device_capability with cuda.is_available()

3512444
Select commit
Loading
Failed to load commit list.