From 568b73fdf86dcc12a88b526633c7bef4bf8cf58f Mon Sep 17 00:00:00 2001 From: Tianlei Wu Date: Thu, 16 Feb 2023 14:57:33 -0800 Subject: [PATCH] Fix stable diffusion onnx pipeline error when batch_size > 1 (#2366) fix safety_checker for batch_size > 1 --- .../stable_diffusion/pipeline_onnx_stable_diffusion.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py b/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py index bbef14ea16..90d48d5330 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py @@ -303,9 +303,6 @@ class OnnxStableDiffusionPipeline(DiffusionPipeline): self.numpy_to_pil(image), return_tensors="np" ).pixel_values.astype(image.dtype) - image, has_nsfw_concepts = self.safety_checker(clip_input=safety_checker_input, images=image) - - # There will throw an error if use safety_checker batchsize>1 images, has_nsfw_concept = [], [] for i in range(image.shape[0]): image_i, has_nsfw_concept_i = self.safety_checker(