return image before postprocess for Dreamshaper RealESRGAN

Signed-off-by: lijian <lijian6@sugon.com>

return image before postprocess for Dreamshaper RealESRGAN
Signed-off-by: lijian <lijian6@sugon.com>
405574bc · lijian6 · d5dcc5c6 · 405574bc
Commit 405574bc authored Aug 08, 2024 by lijian6
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py ...lines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +1 -0

No files found.
--- a/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
+++ b/src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
@@ -1295,6 +1295,7 @@ class StableDiffusionXLPipeline(
        else:
            image = latents

+        return image
        if not output_type == "latent":
            # apply watermark if available
            if self.watermark is not None: