style

Author: yiyixuxu

src/diffusers/pipelines/pag/pipeline_pag_sana.py

@@ -919,9 +919,11 @@ def __call__(
             try:
                 image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
             except torch.cuda.OutOfMemoryError as e:
-                warnings.warn(f"{e}. \n"
-                              f"Try to use VAE tiling for large images. For example: \n"
-                              f"pipe.vae.enable_tiling(tile_sample_min_width=512, tile_sample_min_height=512)")
+                warnings.warn(
+                    f"{e}. \n"
+                    f"Try to use VAE tiling for large images. For example: \n"
+                    f"pipe.vae.enable_tiling(tile_sample_min_width=512, tile_sample_min_height=512)"
+                )
             if use_resolution_binning:
                 image = self.image_processor.resize_and_crop_tensor(image, orig_width, orig_height)
 

src/diffusers/pipelines/sana/pipeline_sana.py

@@ -957,9 +957,11 @@ def __call__(
             try:
                 image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
             except torch.cuda.OutOfMemoryError as e:
-                warnings.warn(f"{e}. \n"
-                              f"Try to use VAE tiling for large images. For example: \n"
-                              f"pipe.vae.enable_tiling(tile_sample_min_width=512, tile_sample_min_height=512)")
+                warnings.warn(
+                    f"{e}. \n"
+                    f"Try to use VAE tiling for large images. For example: \n"
+                    f"pipe.vae.enable_tiling(tile_sample_min_width=512, tile_sample_min_height=512)"
+                )
             if use_resolution_binning:
                 image = self.image_processor.resize_and_crop_tensor(image, orig_width, orig_height)