Skip to content

GitLab

Unverified Commit 1a6def3d authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix post-processing (#2968) 


Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent 0c63c383
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 12 deletions
src/diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
View file @ 1a6def3d
...@@ -734,6 +734,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMixin ...@@ -734,6 +734,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMixin
image = latents image = latents
has_nsfw_concept = None has_nsfw_concept = None
else:
image = self.decode_latents(latents) image = self.decode_latents(latents)
if self.safety_checker is not None: if self.safety_checker is not None:
......
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
View file @ 1a6def3d
...@@ -742,6 +742,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMi ...@@ -742,6 +742,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMi
image = latents image = latents
has_nsfw_concept = None has_nsfw_concept = None
else:
image = self.decode_latents(latents) image = self.decode_latents(latents)
if self.safety_checker is not None: if self.safety_checker 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