From 8ff3ab01a52e5d0eb366a2fbb74fa7fc0f26fc15 Mon Sep 17 00:00:00 2001
From: Mihir Jagtap <mjagtap@wisc.edu>
Date: Thu, 5 Feb 2026 19:32:35 -0800
Subject: [PATCH] Update denoising.py

---
 fastvideo/pipelines/stages/denoising.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/fastvideo/pipelines/stages/denoising.py b/fastvideo/pipelines/stages/denoising.py
index a6c235b5d..d1c0a2cd5 100644
--- a/fastvideo/pipelines/stages/denoising.py
+++ b/fastvideo/pipelines/stages/denoising.py
@@ -176,6 +176,13 @@ class DenoisingStage(PipelineStage):
             },
         )
 
+        camera_kwargs = self.prepare_extra_func_kwargs(
+            self.transformer.forward,
+            {
+                "camera_states": batch.camera_states,
+            },
+        )
+
         # Prepare STA parameters
         if st_attn_available and self.attn_backend == SlidingTileAttentionBackend:
             self.prepare_sta_param(batch, fastvideo_args)
@@ -406,6 +413,7 @@ class DenoisingStage(PipelineStage):
                             **image_kwargs,
                             **pos_cond_kwargs,
                             **action_kwargs,
+                            **camera_kwargs,
                         )
 
                     if batch.do_classifier_free_guidance:
@@ -423,6 +431,7 @@ class DenoisingStage(PipelineStage):
                                 **image_kwargs,
                                 **neg_cond_kwargs,
                                 **action_kwargs,
+                                **camera_kwargs,
                             )
 
                         noise_pred_text = noise_pred