Disable SC

patch_sdxl.py

@@ -1,6 +1,3 @@
-    
-
-
 import inspect
 from typing import Any, Callable, Dict, List, Optional, Union, Tuple
 
@@ -29,7 +26,6 @@ from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOut
 
 
 
-from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 from transformers import CLIPFeatureExtractor
 import numpy as np
 import torch
@@ -40,27 +36,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 torch_device = device
 torch_dtype = torch.float16
 
-safety_checker = StableDiffusionSafetyChecker.from_pretrained(
-    "CompVis/stable-diffusion-safety-checker"
-).to(device)
-feature_extractor = CLIPFeatureExtractor.from_pretrained(
-    "openai/clip-vit-base-patch32"
-)
-
-def check_nsfw_images(
-    images: list[Image.Image],
-) -> list[bool]:
-    safety_checker_input = feature_extractor(images, return_tensors="pt").to(device)
-    images_np = [np.array(img) for img in images]
-
-    _, has_nsfw_concepts = safety_checker(
-        images=images_np,
-        clip_input=safety_checker_input.pixel_values.to(torch_device),
-    )
-    return has_nsfw_concepts
-    
-
-
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
 
@@ -569,12 +544,11 @@ class SDEmb(StableDiffusionXLPipeline):
             # apply watermark if available
             if self.watermark is not None:
                 image = self.watermark.apply_watermark(image)
-
             image = self.image_processor.postprocess(image, output_type=output_type)
-        maybe_nsfw = any(check_nsfw_images(image))
-        if maybe_nsfw:
-            print('This image could be NSFW so we return a blank image.')
-            return StableDiffusionXLPipelineOutput(images=[Image.new('RGB', (1024, 1024))])
+        #maybe_nsfw = any(check_nsfw_images(image))
+        #if maybe_nsfw:
+        #    print('This image could be NSFW so we return a blank image.')
+        #    return StableDiffusionXLPipelineOutput(images=[Image.new('RGB', (1024, 1024))])
 
         # Offload all models
         self.maybe_free_model_hooks()