Spaces:
Sleeping
Sleeping
Yinhong Liu
commited on
Commit
·
5703b3a
1
Parent(s):
67fdb88
fix sd3 pipeline
Browse files- sid/pipeline_sid_sd3.py +2 -0
sid/pipeline_sid_sd3.py
CHANGED
|
@@ -675,6 +675,8 @@ class SiDSD3Pipeline(
|
|
| 675 |
num_images_per_prompt: Optional[int] = 1,
|
| 676 |
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
|
| 677 |
latents: Optional[torch.FloatTensor] = None,
|
|
|
|
|
|
|
| 678 |
output_type: Optional[str] = "pil",
|
| 679 |
return_dict: bool = True,
|
| 680 |
callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
|
|
|
|
| 675 |
num_images_per_prompt: Optional[int] = 1,
|
| 676 |
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
|
| 677 |
latents: Optional[torch.FloatTensor] = None,
|
| 678 |
+
prompt_embeds: Optional[torch.FloatTensor] = None,
|
| 679 |
+
pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
|
| 680 |
output_type: Optional[str] = "pil",
|
| 681 |
return_dict: bool = True,
|
| 682 |
callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
|