fastvideo.pipelines.stages.causal_denoising#
Module Contents#
Classes#
Denoising stage for causal diffusion. |
Data#
API#
- class fastvideo.pipelines.stages.causal_denoising.CausalDMDDenosingStage(transformer, scheduler, transformer_2=None)[source]#
Bases:
fastvideo.pipelines.stages.denoising.DenoisingStageDenoising 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.