Skip to content

Commit 0961a03

Browse files
committed
add post init for safty checker
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 8d415a6 commit 0961a03

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/pipelines/stable_diffusion/safety_checker.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ def __init__(self, config: CLIPConfig):
4646

4747
self.concept_embeds_weights = nn.Parameter(torch.ones(17), requires_grad=False)
4848
self.special_care_embeds_weights = nn.Parameter(torch.ones(3), requires_grad=False)
49+
self.post_init()
4950

5051
@torch.no_grad()
5152
def forward(self, clip_input, images):

0 commit comments

Comments
 (0)