Skip to content

GitLab

Unverified Commit 5a6edac0 authored by Fanli Lin's avatar Fanli Lin Committed by GitHub
Browse files

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

no cuda only
parent e8fc8b1f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/quantization/bnb/test_mixed_int8.py
View file @ 5a6edac0
...@@ -315,7 +315,7 @@ class BnB8bitBasicTests(Base8bitTests): ...@@ -315,7 +315,7 @@ class BnB8bitBasicTests(Base8bitTests):
_ = self.model_fp16.float() _ = self.model_fp16.float()
# Check that this does not throw an error # Check that this does not throw an error
_ = self.model_fp16.cuda() _ = self.model_fp16.to(torch_device)
class Bnb8bitDeviceTests(Base8bitTests): class Bnb8bitDeviceTests(Base8bitTests):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment