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
......@@ -734,6 +734,7 @@ class AltDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMixin
image = latents
has_nsfw_concept = None
else:
image = self.decode_latents(latents)
if self.safety_checker is not None:
......
......@@ -742,6 +742,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline, TextualInversionLoaderMi
image = latents
has_nsfw_concept = None
else:
image = self.decode_latents(latents)
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