tests

Author: sayakpaul

tests/quantization/bnb/test_4bit.py

@@ -485,7 +485,7 @@ def test_moving_to_cpu_throws_warning(self):
 
         assert "Pipelines loaded with `dtype=torch.float16`" in cap_logger.out
 
-    @require_accelerate_version_greater("1.1.0")
+    @require_accelerate_version_greater("1.0.0")
     def test_pipeline_cuda_placement_works_with_nf4(self):
         transformer_nf4_config = BitsAndBytesConfig(
             load_in_4bit=True,

tests/quantization/bnb/test_mixed_int8.py

@@ -434,7 +434,7 @@ def test_generate_quality_dequantize(self):
             output_type="np",
         ).images
 
-    @require_accelerate_version_greater("1.1.0")
+    @require_accelerate_version_greater("1.0.0")
     def test_pipeline_cuda_placement_works_with_mixed_int8(self):
         transformer_8bit_config = BitsAndBytesConfig(load_in_8bit=True)
         transformer_8bit = SD3Transformer2DModel.from_pretrained(