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

@@ -427,7 +427,10 @@ def run_safety_checker(self, image, device, dtype):
         if self.safety_checker :
             has_nsfw_concept = False
         else:
-            feature_extractor_input = self.image_processor.postprocess(image, output_type="pil")
+            if torch.is_tensor(image):
+                feature_extractor_input = self.image_processor.postprocess(image, output_type="pil")
+            else:
+                feature_extractor_input = self.numpy_to_pil(image)
             safety_checker_input = self.feature_extractor(feature_extractor_input, return_tensors="pt").to(device)
             image, has_nsfw_concept = self.safety_checker(
                 images=image, clip_input=safety_checker_input.pixel_values.to(dtype)