Yinhong Liu commited on
Commit
5703b3a
·
1 Parent(s): 67fdb88

fix sd3 pipeline

Browse files
Files changed (1) hide show
  1. 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,