Skip to content

GitLab

Unverified Commit 4c5718a0 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix a bug in StableDiffusionUpscalePipeline.run_safety_checker (#4886) 



fix
Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent 2340ed62
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py
View file @ 4c5718a0
...@@ -165,10 +165,10 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversionLoaderMi ...@@ -165,10 +165,10 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline, TextualInversionLoaderMi
# We'll offload the last model manually. # We'll offload the last model manually.
self.final_offload_hook = hook self.final_offload_hook = hook
# Copied from diffusers.pipelines.deepfloyd_if.pipeline_if.IFPipeline.run_safety_checker
def run_safety_checker(self, image, device, dtype): def run_safety_checker(self, image, device, dtype):
if self.safety_checker is not None: if self.safety_checker is not None:
safety_checker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(device) feature_extractor_input = self.image_processor.postprocess(image, output_type="pil")
safety_checker_input = self.feature_extractor(feature_extractor_input, return_tensors="pt").to(device)
image, nsfw_detected, watermark_detected = self.safety_checker( image, nsfw_detected, watermark_detected = self.safety_checker(
images=image, images=image,
clip_input=safety_checker_input.pixel_values.to(dtype=dtype), clip_input=safety_checker_input.pixel_values.to(dtype=dtype),
......
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