fastvideo.pipelines.stages.encoding

fastvideo.pipelines.stages.encoding#

Encoding stage for diffusion pipelines.

Module Contents#

Classes#

EncodingStage

Stage for encoding pixel space representations into latent space.

Data#

logger

API#

class fastvideo.pipelines.stages.encoding.EncodingStage(vae: fastvideo.models.vaes.common.ParallelTiledVAE)[source]#

Bases: fastvideo.pipelines.stages.base.PipelineStage

Stage for encoding pixel space representations into latent space.

This stage handles the encoding of pixel-space video/images into latent representations for further processing in the diffusion pipeline.

Initialization

forward(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) → fastvideo.pipelines.pipeline_batch_info.ForwardBatch[source]#

Encode pixel space representations into latent space.

Parameters:

  • batch – The current batch information.

  • fastvideo_args – The inference arguments.

Returns:

The batch with encoded latents.

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

Verify encoding stage inputs.

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

Verify encoding stage outputs.

fastvideo.pipelines.stages.encoding.logger[source]#

‘init_logger(…)’

Contents