From 0961a03f184ca61f4c9422fb180c1297335c075c Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Fri, 5 Dec 2025 10:27:53 +0000 Subject: [PATCH 1/3] 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): From 80df03d1883792f48cb97f58adccaab5234399b2 Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Mon, 8 Dec 2025 12:54:02 +0000 Subject: [PATCH 2/3] check transformers version before post init Signed-off-by: jiqing-feng --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 0a746fff172b..453550ce14fb 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -17,7 +17,7 @@ import torch.nn as nn from transformers import CLIPConfig, CLIPVisionModel, PreTrainedModel -from ...utils import logging +from ...utils import logging, is_transformers_version logger = logging.get_logger(__name__) @@ -46,7 +46,9 @@ 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() + # Model requires post_init after transformers v4.57.3 + if is_transformers_version(">", "4.57.3"): + self.post_init() @torch.no_grad() def forward(self, clip_input, images): From 312e0fb81b218d13db4f875a263ea037fe30b8ce Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Mon, 8 Dec 2025 05:36:16 +0000 Subject: [PATCH 3/3] Apply style fixes --- src/diffusers/pipelines/stable_diffusion/safety_checker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/stable_diffusion/safety_checker.py b/src/diffusers/pipelines/stable_diffusion/safety_checker.py index 453550ce14fb..65daafe01237 100644 --- a/src/diffusers/pipelines/stable_diffusion/safety_checker.py +++ b/src/diffusers/pipelines/stable_diffusion/safety_checker.py @@ -17,7 +17,7 @@ import torch.nn as nn from transformers import CLIPConfig, CLIPVisionModel, PreTrainedModel -from ...utils import logging, is_transformers_version +from ...utils import is_transformers_version, logging logger = logging.get_logger(__name__)