fastvideo.pipelines.stages.image_encoding

Image encoding stages for I2V diffusion pipelines.

This module contains implementations of image encoding stages for diffusion pipelines.

Module Contents

Classes

ImageEncodingStage

Stage for encoding image prompts into embeddings for diffusion models.

Data

logger

API

class fastvideo.pipelines.stages.image_encoding.ImageEncodingStage(image_encoder, image_processor)

Bases: fastvideo.pipelines.stages.base.PipelineStage

Stage for encoding image prompts into embeddings for diffusion models.

This stage handles the encoding of image prompts into the embedding space expected by the diffusion model.

Initialization

Initialize the prompt encoding stage.

Parameters:

• enable_logging – Whether to enable logging for this stage.

• is_secondary – Whether this is a secondary image encoder.

forward(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) → fastvideo.pipelines.pipeline_batch_info.ForwardBatch

Encode the prompt into image encoder hidden states.

Parameters:

• batch – The current batch information.

• fastvideo_args – The inference arguments.

Returns:

The batch with encoded prompt embeddings.

verify_input(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) → fastvideo.pipelines.stages.validators.VerificationResult

Verify image encoding stage inputs.

verify_output(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) → fastvideo.pipelines.stages.validators.VerificationResult

Verify image encoding stage outputs.

fastvideo.pipelines.stages.image_encoding.logger

‘init_logger(…)’