fastvideo.v1.pipelines.stages.conditioning

Conditioning stage for diffusion pipelines.

Module Contents

Classes

ConditioningStage

Stage for applying conditioning to the diffusion process.

Data

logger

API

class fastvideo.v1.pipelines.stages.conditioning.ConditioningStage

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

Stage for applying conditioning to the diffusion process.

This stage handles the application of conditioning, such as classifier-free guidance, to the diffusion process.

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

Apply conditioning to the diffusion process.

Parameters:

• batch – The current batch information.

• fastvideo_args – The inference arguments.

Returns:

The batch with applied conditioning.

fastvideo.v1.pipelines.stages.conditioning.logger

‘init_logger(…)’