fastvideo.v1.pipelines.stages.input_validation

Input validation stage for diffusion pipelines.

Module Contents

Classes

InputValidationStage

Stage for validating and preparing inputs for diffusion pipelines.

Data

logger

API

class fastvideo.v1.pipelines.stages.input_validation.InputValidationStage

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

Stage for validating and preparing inputs for diffusion pipelines.

This stage validates that all required inputs are present and properly formatted before proceeding with 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

Validate and prepare inputs.

Parameters:

• batch – The current batch information.

• fastvideo_args – The inference arguments.

Returns:

The validated batch information.

fastvideo.v1.pipelines.stages.input_validation.logger

‘init_logger(…)’