fix safety check (#217)

89e95210 · Suraj Patil · GitHub · 65ea7d6b · 89e95210
Unverified Commit 89e95210 authored Aug 19, 2022 by Suraj Patil Committed by GitHub Aug 19, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/diffusers/pipelines/stable_diffusion/safety_checker.py src/diffusers/pipelines/stable_diffusion/safety_checker.py +1 -1

No files found.
--- 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: