Update pipline_StableDiffusion_ConsistentID.py

pipline_StableDiffusion_ConsistentID.py

@@ -419,6 +419,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
         class_tokens_mask: Optional[torch.LongTensor] = None,
         prompt_embeds_text_only: Optional[torch.FloatTensor] = None,
         retouching: bool=False,
+        need_safetycheck: bool=True,
     ):
         # 0. Default height and width to unet
         height = height or self.unet.config.sample_size * self.vae_scale_factor
@@ -604,9 +605,12 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
             image = self.decode_latents(latents)
 
             # 9.2 Run safety checker
-            image, has_nsfw_concept = self.run_safety_checker(
-                image, device, prompt_embeds.dtype
-            )
+            if need_safetycheck:
+                image, has_nsfw_concept = self.run_safety_checker(
+                    image, device, prompt_embeds.dtype
+                )
+            else:
+                has_nsfw_concept = None
 
             # 9.3 Convert to PIL
             image = self.numpy_to_pil(image)