make

Author: hlky

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

@@ -908,8 +908,16 @@ def __call__(
 
         # 0. Default height and width to unet
         if not height or not width:
-            height = self.unet.config.sample_size if self._is_unet_config_sample_size_int else self.unet.config.sample_size[0]
-            width = self.unet.config.sample_size if self._is_unet_config_sample_size_int else self.unet.config.sample_size[1]
+            height = (
+                self.unet.config.sample_size
+                if self._is_unet_config_sample_size_int
+                else self.unet.config.sample_size[0]
+            )
+            width = (
+                self.unet.config.sample_size
+                if self._is_unet_config_sample_size_int
+                else self.unet.config.sample_size[1]
+            )
             height, width = height * self.vae_scale_factor, width * self.vae_scale_factor
         # to deal with lora scaling and other possible forward hooks
 

tests/pipelines/stable_diffusion/test_stable_diffusion.py

@@ -839,7 +839,7 @@ def callback_on_step_end(pipe, i, t, callback_kwargs):
         # compare the intermediate latent to the output of the interrupted process
         # they should be the same
         assert torch.allclose(intermediate_latent, output_interrupted, atol=1e-4)
-        
+
     def test_pipeline_accept_tuple_type_unet_sample_size(self):
         # the purpose of this test is to see whether the pipeline would accept a unet with the tuple-typed sample size
         sd_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"