Skip to content

Commit a093626

Browse files
suzukimainhlky
andauthored
Update src/diffusers/pipelines/stable_diffusion/safety_checker.py
Co-authored-by: hlky <[email protected]>
1 parent 190a641 commit a093626

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/stable_diffusion/safety_checker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def forward(self, clip_input, images):
103103
result_img["special_scores"][concept_idx] = round(concept_cos - concept_threshold + self.adjustment, 3)
104104
if result_img["special_scores"][concept_idx] > 0:
105105
result_img["special_care"].append({concept_idx, result_img["special_scores"][concept_idx]})
106-
self.adjustment = 0.01
106+
self.adjustment += 0.01
107107

108108
for concept_idx in range(len(cos_dist[0])):
109109
concept_cos = cos_dist[i][concept_idx]

0 commit comments

Comments
 (0)