Skip to content

GitLab

Commit 405574bc authored by lijian6's avatar lijian6
Browse files

return image before postprocess for Dreamshaper RealESRGAN 


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent d5dcc5c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py
View file @ 405574bc
...@@ -1295,6 +1295,7 @@ class StableDiffusionXLPipeline( ...@@ -1295,6 +1295,7 @@ class StableDiffusionXLPipeline(
else: else:
image = latents image = latents
return image
if not output_type == "latent": if not output_type == "latent":
# apply watermark if available # apply watermark if available
if self.watermark is not None: if self.watermark is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment