quality

Author: sayakpaul

tests/quantization/gguf/test_gguf.py

@@ -8,14 +8,14 @@
 from diffusers import (
     AuraFlowPipeline,
     AuraFlowTransformer2DModel,
+    DiffusionPipeline,
     FluxControlPipeline,
     FluxPipeline,
     FluxTransformer2DModel,
     GGUFQuantizationConfig,
     HiDreamImageTransformer2DModel,
     SD3Transformer2DModel,
     StableDiffusion3Pipeline,
-    DiffusionPipeline,
 )
 from diffusers.utils import load_image
 from diffusers.utils.testing_utils import (
@@ -29,13 +29,15 @@
     numpy_cosine_similarity_distance,
     require_accelerate,
     require_big_accelerator,
-    require_torch_version_greater,
     require_gguf_version_greater_or_equal,
     require_peft_backend,
+    require_torch_version_greater,
     torch_device,
 )
+
 from ..test_torch_compile_utils import QuantCompileTests
 
+
 if is_gguf_available():
     from diffusers.quantizers.gguf.utils import GGUFLinear, GGUFParameter
 
@@ -592,9 +594,7 @@ def _init_pipeline(self, *args, **kwargs):
             self.gguf_ckpt, quantization_config=self.quantization_config, torch_dtype=self.torch_dtype
         )
         pipe = DiffusionPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-3.5-medium",
-            transformer=transformer,
-            torch_dtype=self.torch_dtype
+            "stabilityai/stable-diffusion-3.5-medium", transformer=transformer, torch_dtype=self.torch_dtype
         )
         return pipe