8000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 352a230 commit 70e737bCopy full SHA for 70e737b
tests/bnb/test_bnb_regression.py
@@ -114,7 +114,7 @@ def test_opt_350m_4bit_double_quant():
114
torch.testing.assert_allclose(output, expected)
115
116
117
-@pytest.mark.skipif(torch.cuda.is_available(), reason="No CUDA device available.")
+@pytest.mark.skipif(not torch.cuda.is_available(), reason="No CUDA device available.")
118
def test_opt_350m_4bit_compute_dtype_float16():
119
torch.manual_seed(0)
120
bnb_config = BitsAndBytesConfig(
0 commit comments