Spaces:
Build error
Build error
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
|