JackAILab commited on
Commit
5043913
1 Parent(s): 0b86882

Update pipline_StableDiffusion_ConsistentID.py

Browse files
pipline_StableDiffusion_ConsistentID.py CHANGED
@@ -418,6 +418,7 @@ class ConsistentIDStableDiffusionPipeline(StableDiffusionPipeline):
418
  start_merge_step: int = 0,
419
  class_tokens_mask: Optional[torch.LongTensor] = None,
420
  prompt_embeds_text_only: Optional[torch.FloatTensor] = None,
 
421
  ):
422
  # 0. Default height and width to unet
423
  height = height or self.unet.config.sample_size * self.vae_scale_factor
 
418
  start_merge_step: int = 0,
419
  class_tokens_mask: Optional[torch.LongTensor] = None,
420
  prompt_embeds_text_only: Optional[torch.FloatTensor] = None,
421
+ retouching: bool=False,
422
  ):
423
  # 0. Default height and width to unet
424
  height = height or self.unet.config.sample_size * self.vae_scale_factor