[tests] no hard-coded cuda (#11186)

no cuda only

[tests] no hard-coded cuda (#11186)
no cuda only
5a6edac0 · Fanli Lin · GitHub · e8fc8b1f · 5a6edac0
Unverified Commit 5a6edac0 authored Apr 01, 2025 by Fanli Lin Committed by GitHub Apr 01, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/quantization/bnb/test_mixed_int8.py tests/quantization/bnb/test_mixed_int8.py +1 -1

No files found.
--- a/tests/quantization/bnb/test_mixed_int8.py
+++ b/tests/quantization/bnb/test_mixed_int8.py
@@ -315,7 +315,7 @@ class BnB8bitBasicTests(Base8bitTests):
        _ = self.model_fp16.float()

        # Check that this does not throw an error
-        _ = self.model_fp16.cuda()
+        _ = self.model_fp16.to(torch_device)


 class Bnb8bitDeviceTests(Base8bitTests):