Skip to content

Commit 568b73f

Browse files
authored
Fix stable diffusion onnx pipeline error when batch_size > 1 (huggingface#2366)
fix safety_checker for batch_size > 1
1 parent 8e1cae5 commit 568b73f

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -303,9 +303,6 @@ def __call__(
303303
self.numpy_to_pil(image), return_tensors="np"
304304
).pixel_values.astype(image.dtype)
305305

306-
image, has_nsfw_concepts = self.safety_checker(clip_input=safety_checker_input, images=image)
307-
308-
# There will throw an error if use safety_checker batchsize>1
309306
images, has_nsfw_concept = [], []
310307
for i in range(image.shape[0]):
311308
image_i, has_nsfw_concept_i = self.safety_checker(

0 commit comments

Comments
 (0)