Skip to content

Commit 52e4e0e

Browse files
committed
no torch_gpu
1 parent a7e9f85 commit 52e4e0e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_modeling_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1731,7 +1731,7 @@ def tearDown(self):
17311731
gc.collect()
17321732
backend_empty_cache(torch_device)
17331733

1734-
@require_torch_gpu
1734+
@require_torch_accelerator
17351735
@require_torch_2
17361736
@is_torch_compile
17371737
@slow

0 commit comments

Comments
 (0)