fastvideo.pipelines.stages.causal_denoising
#
Module Contents#
Classes#
Denoising stage for causal diffusion. |
Data#
API#
- class fastvideo.pipelines.stages.causal_denoising.CausalDMDDenosingStage(transformer, scheduler)[source]#
Bases:
fastvideo.pipelines.stages.denoising.DenoisingStage
Denoising stage for causal diffusion.
Initialization
- forward(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) fastvideo.pipelines.pipeline_batch_info.ForwardBatch [source]#
- verify_input(batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) fastvideo.pipelines.stages.validators.VerificationResult [source]#
Verify denoising stage inputs.