Update pipline_StableDiffusionXL_ConsistentID.py

pipline_StableDiffusionXL_ConsistentID.py

@@ -464,6 +464,8 @@ class ConsistentIDStableDiffusionXLPipeline(StableDiffusionXLPipeline):
         start_merge_step: int = 0,
         class_tokens_mask: Optional[torch.LongTensor] = None,
         prompt_embeds_text_only: Optional[torch.FloatTensor] = None,
+        retouching: bool=False,
+        need_safetycheck: bool=True,
         ### add for sdxl
         negative_prompt_2: Optional[Union[str, List[str]]] = None,   
         prompt_2: Optional[Union[str, List[str]]] = None,     
@@ -694,7 +696,7 @@ class ConsistentIDStableDiffusionXLPipeline(StableDiffusionXLPipeline):
 
         if not output_type == "latent":
             image = self.vae.decode(latents / self.vae.config.scaling_factor, return_dict=False)[0]
-        else:
+        else: ### TODO add self.run_safety_checker (if need_safetycheck True)
             image = latents
             return StableDiffusionXLPipelineOutput(images=image)