fastvideo.v1.pipelines.wan.wan_pipeline#

Wan video diffusion pipeline implementation.

This module contains an implementation of the Wan video diffusion pipeline using the modular pipeline architecture.

Module Contents#

Classes#

WanPipeline

WanValidationPipeline

Validation pipeline for Wan2.1, assumes that the input are preprocess latents.

Data#

API#

fastvideo.v1.pipelines.wan.wan_pipeline.EntryClass#

None

class fastvideo.v1.pipelines.wan.wan_pipeline.WanPipeline(model_path: str, fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs, config: Optional[Dict[str, Any]] = None, required_config_modules: Optional[List[str]] = None)[source]#

Bases: fastvideo.v1.pipelines.composed_pipeline_base.ComposedPipelineBase

create_pipeline_stages(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs)[source]#

Set up pipeline stages with proper dependency injection.

initialize_pipeline(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs)[source]#
class fastvideo.v1.pipelines.wan.wan_pipeline.WanValidationPipeline(model_path: str, fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs, config: Optional[Dict[str, Any]] = None, required_config_modules: Optional[List[str]] = None)[source]#

Bases: fastvideo.v1.pipelines.composed_pipeline_base.ComposedPipelineBase

Validation pipeline for Wan2.1, assumes that the input are preprocess latents.

Initialization

Initialize the pipeline. After init, the pipeline should be ready to use. The pipeline should be stateless and not hold any batch state.

create_pipeline_stages(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs)[source]#

Set up pipeline stages with proper dependency injection.

initialize_pipeline(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs)[source]#
fastvideo.v1.pipelines.wan.wan_pipeline.logger[source]#

β€˜init_logger(…)’