diff --git a/tests/quantization/bnb/test_4bit.py b/tests/quantization/bnb/test_4bit.py index 06116cac3adc..98005cfbc810 100644 --- a/tests/quantization/bnb/test_4bit.py +++ b/tests/quantization/bnb/test_4bit.py @@ -872,6 +872,7 @@ def test_fp4_double_safe(self): @require_torch_version_greater("2.7.1") +@require_bitsandbytes_version_greater("0.45.5") class Bnb4BitCompileTests(QuantCompileTests): @property def quantization_config(self): diff --git a/tests/quantization/bnb/test_mixed_int8.py b/tests/quantization/bnb/test_mixed_int8.py index 2ea4cdfde870..f3bbc34e8b2c 100644 --- a/tests/quantization/bnb/test_mixed_int8.py +++ b/tests/quantization/bnb/test_mixed_int8.py @@ -837,6 +837,7 @@ def test_serialization_sharded(self): @require_torch_version_greater_equal("2.6.0") +@require_bitsandbytes_version_greater("0.45.5") class Bnb8BitCompileTests(QuantCompileTests): @property def quantization_config(self):