From 2d395424eaaa9a06a794906f8e8d325f4c850376 Mon Sep 17 00:00:00 2001 From: Acly Date: Thu, 5 Feb 2026 16:28:32 +0100 Subject: [PATCH] Fix nsfw filter with transformers>5 --- nsfw.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/nsfw.py b/nsfw.py index 3bbd659..056bc18 100644 --- a/nsfw.py +++ b/nsfw.py @@ -1,5 +1,4 @@ from __future__ import annotations -from weakref import ref as WeakRef from pathlib import Path from tqdm import tqdm import torch @@ -39,6 +38,10 @@ class CLIPSafetyChecker(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 hasattr(self, "post_init"): + self.post_init() + def forward(self, clip_input, images: Tensor, sensitivity: float): with torch.no_grad(): image_batch = self.vision_model(clip_input)[1]