fastvideo.pipelines.stages.causal_denoising

fastvideo.pipelines.stages.causal_denoising#

Module Contents#

Classes#

CausalDMDDenosingStage

Denoising stage for causal diffusion.

Data#

logger

API#

class fastvideo.pipelines.stages.causal_denoising.CausalDMDDenosingStage(transformer, scheduler, transformer_2=None)[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.

fastvideo.pipelines.stages.causal_denoising.logger[source]#

‘init_logger(…)’

Contents