Skip to content

Commit 5f1407d

Browse files
linoytsabanhlky
andauthored
Update examples/community/pipeline_flux_rf_inversion.py
Co-authored-by: hlky <[email protected]>
1 parent 074cf5a commit 5f1407d

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

examples/community/pipeline_flux_rf_inversion.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -678,6 +678,12 @@ def __call__(
678678
prompt_2 (`str` or `List[str]`, *optional*):
679679
The prompt or prompts to be sent to `tokenizer_2` and `text_encoder_2`. If not defined, `prompt` is
680680
will be used instead
681+
inverted_latents (`torch.Tensor`, *optional*):
682+
The inverted latents from `pipe.invert`.
683+
image_latents (`torch.Tensor`, *optional*):
684+
The image latents from `pipe.invert`.
685+
latent_image_ids (`torch.Tensor`, *optional*):
686+
The latent image ids from `pipe.invert`.
681687
height (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
682688
The height in pixels of the generated image. This is set to 1024 by default for the best results.
683689
width (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):

0 commit comments

Comments
 (0)