AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

add post init for safty checker (#12794)

Browse Source 
* add post init for safty checker

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* check transformers version before post init

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>

* Apply style fixes

---------

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
jiqing-feng
2025-12-08 14:01:03 +08:00
committed by GitHub
parent 5a74319715
commit f67639b0bb
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/pipelines/stable_diffusion/safety_checker.py
View File

@@ -17,7 +17,7 @@ import torch
import torch.nn as nn
from transformers import CLIPConfig, CLIPVisionModel, PreTrainedModel
from ...utils import logging
from ...utils import is_transformers_version, logging
logger = logging.get_logger(__name__)
@@ -46,6 +46,9 @@ class StableDiffusionSafetyChecker(PreTrainedModel):
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)
# 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):