From e7b69cbe191ede2697f92b953caf54e0599f3ede Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 21 Aug 2022 15:29:10 +0000 Subject: [PATCH] [Safety Checker] Lower adjustment value --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 3c43d7ffd9..5a315c3367 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -43,7 +43,10 @@ class StableDiffusionSafetyChecker(PreTrainedModel): batch_size = image_embeds.shape[0] for i in range(batch_size): result_img = {"special_scores": {}, "special_care": [], "concept_scores": {}, "bad_concepts": []} - adjustment = 0.05 + + # increase this value to create a stronger `nfsw` filter + # at the cost of increasing the possibility of filtering benign images + adjustment = 0.0 for concet_idx in range(len(special_cos_dist[0])): concept_cos = special_cos_dist[i][concet_idx]