Skip to content

Commit 51fc901

Browse files
committed
Update threshold dictionary
1 parent 3c734d4 commit 51fc901

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/diffusers/pipelines/stable_diffusion/safety_checker.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -55,11 +55,11 @@ def update_safety_checker_Level(self, Level):
5555
Level (`int` or `float` or one of the following [`WEAK`], [`MEDIUM`], [`NOMAL`], [`STRONG`], [`MAX`])
5656
"""
5757
Level_dict = {
58-
"WEAK": -1.0,
59-
"MEDIUM": -0.5,
60-
"NOMAL": 0.0,
61-
"STRONG": 0.5,
62-
"MAX": 1.0,
58+
"WEAK": -0.10,
59+
"MEDIUM": -0.01,
60+
"NOMAL": 0.00,
61+
"STRONG": 0.01,
62+
"MAX": 0.10,
6363
}
6464
if Level in Level_dict:
6565
Level = Level_dict[Level]

0 commit comments

Comments
 (0)