fastvideo.v1.pipelines.stages.timestep_preparation

Timestep preparation stages for diffusion pipelines.

This module contains implementations of timestep preparation stages for diffusion pipelines.

Module Contents

Classes

TimestepPreparationStage

Stage for preparing timesteps for the diffusion process.

Data

logger

API

class fastvideo.v1.pipelines.stages.timestep_preparation.TimestepPreparationStage(scheduler)

Bases: fastvideo.v1.pipelines.stages.base.PipelineStage

Stage for preparing timesteps for the diffusion process.

This stage handles the preparation of the timestep sequence that will be used during the diffusion process.

Initialization

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

Prepare timesteps for the diffusion process.

Parameters:

• batch – The current batch information.

• fastvideo_args – The inference arguments.

Returns:

The batch with prepared timesteps.

fastvideo.v1.pipelines.stages.timestep_preparation.logger

‘init_logger(…)’