From 89e9521048067acacfdcbc2b985af8f6b155cfb6 Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Fri, 19 Aug 2022 18:04:58 +0530 Subject: [PATCH] fix safety check (#217) --- 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 1c5db4210d..3c43d7ffd9 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -62,7 +62,7 @@ class StableDiffusionSafetyChecker(PreTrainedModel): 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: