Spaces:
Running
on
Zero
Running
on
Zero
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
|