Skip to content

Commit 6939fd4

Browse files
Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
Co-authored-by: Patrick von Platen <[email protected]>
1 parent 89669dc commit 6939fd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ def _encode_prompt(
424424
return prompt_embeds
425425

426426
def run_safety_checker(self, image, device, dtype):
427-
if self.safety_checker is None or output_type == "latent":
427+
if self.safety_checker :
428428
has_nsfw_concept = False
429429
else:
430430
feature_extractor_input = self.image_processor.postprocess(image, output_type="pil")

0 commit comments

Comments
 (0)