Skip to content

Commit f98dfbd

Browse files
committed
annotate compilation test classes with bnb
1 parent 8adc600 commit f98dfbd

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

tests/quantization/bnb/test_4bit.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -862,6 +862,7 @@ def test_fp4_double_safe(self):
862862

863863

864864
@require_torch_version_greater("2.7.1")
865+
@require_bitsandbytes_version_greater("0.45.5")
865866
class Bnb4BitCompileTests(QuantCompileTests):
866867
quantization_config = PipelineQuantizationConfig(
867868
quant_backend="bitsandbytes_8bit",

tests/quantization/bnb/test_mixed_int8.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -827,6 +827,7 @@ def test_serialization_sharded(self):
827827

828828

829829
@require_torch_version_greater_equal("2.6.0")
830+
@require_bitsandbytes_version_greater("0.45.5")
830831
class Bnb8BitCompileTests(QuantCompileTests):
831832
quantization_config = PipelineQuantizationConfig(
832833
quant_backend="bitsandbytes_8bit",

0 commit comments

Comments
 (0)