Skip to content

vllm_omni.diffusion.media

DiffusionMediaOutput dataclass

prepared_for_transport class-attribute instance-attribute 

prepared_for_transport: bool = False

video instance-attribute 

video: VideoMediaOutput

to_cpu 

to_cpu() -> DiffusionMediaOutput

validate 

validate() -> None

with_video 

with_video(video: VideoMediaOutput) -> DiffusionMediaOutput

FloatVideoConsumer

Bases: str, Enum

FRAME_INTERPOLATION class-attribute instance-attribute 

FRAME_INTERPOLATION = 'frame_interpolation'

VIDEO_GUARDRAILS class-attribute instance-attribute 

VIDEO_GUARDRAILS = 'video_guardrails'

VideoColorModel

Bases: str, Enum

RGB class-attribute instance-attribute 

RGB = 'rgb'

VideoMediaOutput dataclass

constraints class-attribute instance-attribute 

constraints: VideoTransportConstraints = field(
    default_factory=VideoTransportConstraints
)

schema_version class-attribute instance-attribute 

schema_version: Literal[1] = 1

spec instance-attribute 

spec: VideoTensorSpec

tensor instance-attribute 

tensor: Tensor

to_cpu 

to_cpu() -> VideoMediaOutput

validate 

validate() -> None

with_tensor 

with_tensor(tensor: Tensor) -> VideoMediaOutput

VideoTensorEncoding

Bases: str, Enum

NORMALIZED_FLOAT class-attribute instance-attribute 

NORMALIZED_FLOAT = 'normalized_float'

UINT8_FRAMES class-attribute instance-attribute 

UINT8_FRAMES = 'uint8_frames'

VideoTensorLayout

Bases: str, Enum

BCTHW class-attribute instance-attribute 

BCTHW = 'bcthw'

BTHWC class-attribute instance-attribute 

BTHWC = 'bthwc'

VideoTensorSpec dataclass

color_model class-attribute instance-attribute 

color_model: VideoColorModel = VideoColorModel.RGB

encoding instance-attribute 

encoding: VideoTensorEncoding

layout instance-attribute 

layout: VideoTensorLayout

value_range instance-attribute 

value_range: VideoValueRange

VideoTransportConstraints dataclass

pending_float_consumers class-attribute instance-attribute 

pending_float_consumers: frozenset[FloatVideoConsumer] = (
    frozenset()
)

VideoValueRange

Bases: str, Enum

NEGATIVE_ONE_TO_ONE class-attribute instance-attribute 

NEGATIVE_ONE_TO_ONE = 'negative_one_to_one'

ZERO_TO_255 class-attribute instance-attribute 

ZERO_TO_255 = 'zero_to_255'

ZERO_TO_ONE class-attribute instance-attribute 

ZERO_TO_ONE = 'zero_to_one'

ensure_request_owned_tensor 

ensure_request_owned_tensor(tensor: Tensor) -> Tensor

slice_diffusion_media_output 

slice_diffusion_media_output(
    media: DiffusionMediaOutput, start: int, stop: int
) -> DiffusionMediaOutput