fastvideo.pipelines.stages.encoding

fastvideo.pipelines.stages.encoding#

Encoding stage for diffusion pipelines.

Module Contents#

Classes#

EncodingStage

Stage for encoding pixel 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 representations into latent space.

This stage handles the encoding of pixel representations into the final input format (e.g., latents).

Initialization

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

Encode pixel representations into latent space.

Parameters:

  • batch – The current batch information.

  • fastvideo_args – The inference arguments.

Returns:

The batch with encoded outputs.

preprocess(image: PIL.Image.Image, vae_scale_factor: int, height: int | None = None, width: int | None = None, resize_mode: str = 'default') → torch.Tensor[source]#
retrieve_latents(encoder_output: torch.Tensor, generator: torch.Generator | None = None, sample_mode: str = 'sample')[source]#
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