Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

Co-authored-by: Patrick von Platen <[email protected]>

Author: yiyixuxu

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

@@ -706,7 +706,9 @@ def __call__(
         if not output_type == "latent":
             image = self.vae.decode(latents / self.vae.config.scaling_factor).sample
 
-        image, has_nsfw_concept = self.run_safety_checker(image, device, prompt_embeds.dtype, output_type=output_type)
+             image, has_nsfw_concept = self.run_safety_checker(image, device, prompt_embeds.dtype)
+         else:
+             has_nsfw_concept = None
 
         do_normalize = [not has_nsfw for has_nsfw in has_nsfw_concept]
         image = self.image_processor.postprocess(image, output_type=output_type, do_normalize=do_normalize)