From 5529445d48bc63f65c07f71ffd0226ca3d6e2cfb Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Fri, 19 Aug 2022 16:59:12 +0530 Subject: [PATCH] fix safety check --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 1c5db4210ddf..3c43d7ffd988 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -62,7 +62,7 @@ def forward(self, clip_input, images): result.append(result_img) - has_nsfw_concepts = [len(result[i]["bad_concepts"]) > 0 or i in range(len(result))] + has_nsfw_concepts = [len(res["bad_concepts"]) > 0 for res in result] for idx, has_nsfw_concept in enumerate(has_nsfw_concepts): if has_nsfw_concept: