diff --git a/tests/quantization/bnb/test_4bit.py b/tests/quantization/bnb/test_4bit.py index c5497d1c8d16..5a0aa89a55a9 100644 --- a/tests/quantization/bnb/test_4bit.py +++ b/tests/quantization/bnb/test_4bit.py @@ -865,6 +865,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 383cdd6849ea..2cf01e4918e4 100644 --- a/tests/quantization/bnb/test_mixed_int8.py +++ b/tests/quantization/bnb/test_mixed_int8.py @@ -830,6 +830,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):