small fix

2025-04-15 20:13:21 +00:00
parent ec53a4dce7
commit 92fbe6e745
1 changed files with 1 additions and 1 deletions
@@ -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: