Skip to content

vllm_omni.diffusion.models.flux.flux_pipeline_mixin ¶

Flux Pipeline Mixin - Shared methods for Flux pipelines.

FluxPipelineMixin ¶

Shared methods for Flux pipelines (FluxPipeline, FluxKontextPipeline).

calculate_shift `staticmethod` ¶

calculate_shift(
    image_seq_len,
    base_seq_len: int = 256,
    max_seq_len: int = 4096,
    base_shift: float = 0.5,
    max_shift: float = 1.15,
)

retrieve_latents `staticmethod` ¶

retrieve_latents(
    encoder_output: Tensor,
    generator: Generator | None = None,
    sample_mode: str = "sample",
)

retrieve_timesteps `staticmethod` ¶

retrieve_timesteps(
    scheduler,
    num_inference_steps: int | None = None,
    device: device | None = None,
    timesteps: list[int] | None = None,
    sigmas: list[float] | None = None,
    **kwargs,
)