Update lcm/lcm_i2i_pipeline.py

lcm/lcm_i2i_pipeline.py

@@ -49,9 +49,11 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
         tokenizer: CLIPTokenizer,
         unet: UNet2DConditionModel,
         scheduler: "LCMSchedulerWithTimestamp",
-        safety_checker: StableDiffusionSafetyChecker,
+        #safety_checker: StableDiffusionSafetyChecker,
+        safety_checker=None,  # Disable NSFW filter
+        requires_safety_checker=False,  # Skip warning
         feature_extractor: CLIPImageProcessor,
-        requires_safety_checker: bool = False,
+        #requires_safety_checker: bool = False,
     ):
         super().__init__()
 
@@ -69,8 +71,10 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
             tokenizer=tokenizer,
             unet=unet,
             scheduler=scheduler,
-            safety_checker=safety_checker,
-            feature_extractor=feature_extractor,
+            safety_checker=None,  # Disable NSFW filter
+            requires_safety_checker=False,  # Skip warning
+            #safety_checker=safety_checker,
+            #feature_extractor=feature_extractor,
         )
         self.vae_scale_factor = 2 ** (len(self.vae.config.block_out_channels) - 1)
         self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)