From 0961a03f184ca61f4c9422fb180c1297335c075c Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Fri, 5 Dec 2025 10:27:53 +0000 Subject: [PATCH] add post init for safty checker Signed-off-by: jiqing-feng --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 16aff102599c..0a746fff172b 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -46,6 +46,7 @@ def __init__(self, config: CLIPConfig): self.concept_embeds_weights = nn.Parameter(torch.ones(17), requires_grad=False) self.special_care_embeds_weights = nn.Parameter(torch.ones(3), requires_grad=False) + self.post_init() @torch.no_grad() def forward(self, clip_input, images):