no cuda only

faaany · faaany · commit fae32ff4d3a4 · 2025-04-01T03:50:33.000-07:00
diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py
@@ -312,7 +312,7 @@ def test_device_and_dtype_assignment(self):
         _ = 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):
