From 92fbe6e745eaabca018eb068a9d8a5929eb92032 Mon Sep 17 00:00:00 2001 From: JerryZhou54 Date: Tue, 15 Apr 2025 20:07:49 +0000 Subject: [PATCH] small fix --- fastvideo/v1/pipelines/stages/denoising.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastvideo/v1/pipelines/stages/denoising.py b/fastvideo/v1/pipelines/stages/denoising.py index 9af18431e..23e1c9903 100644 --- a/fastvideo/v1/pipelines/stages/denoising.py +++ b/fastvideo/v1/pipelines/stages/denoising.py @@ -193,7 +193,7 @@ class DenoisingStage(PipelineStage): attn_metadata = self.attn_metadata_builder.build( current_timestep=i, forward_batch=batch, - inference_args=fastvideo_args, + fastvideo_args=fastvideo_args, ) assert attn_metadata is not None, "attn_metadata cannot be None" else: