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 8adc600 commit f98dfbdCopy full SHA for f98dfbd
tests/quantization/bnb/test_4bit.py
@@ -862,6 +862,7 @@ def test_fp4_double_safe(self):
862
863
864
@require_torch_version_greater("2.7.1")
865
+@require_bitsandbytes_version_greater("0.45.5")
866
class Bnb4BitCompileTests(QuantCompileTests):
867
quantization_config = PipelineQuantizationConfig(
868
quant_backend="bitsandbytes_8bit",
tests/quantization/bnb/test_mixed_int8.py
@@ -827,6 +827,7 @@ def test_serialization_sharded(self):
827
828
829
@require_torch_version_greater_equal("2.6.0")
830
831
class Bnb8BitCompileTests(QuantCompileTests):
832
833
0 commit comments