Update app.py

app.py

@@ -66,7 +66,6 @@ from ltx_pipelines.utils.helpers import (
     simple_denoising_func,
 )
 from ltx_pipelines.utils.media_io import decode_audio_from_file, encode_video
-from ltx_pipelines.utils.denoisers import SimpleDenoiser
 from ltx_pipelines.utils.types import ModalitySpec
 from ltx_core.loader.primitives import LoraPathStrengthAndSDOps
 from ltx_core.loader.sd_ops import LTXV_LORA_COMFY_RENAMING_MAP
@@ -206,7 +205,11 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
         )
 
         video_state, audio_state = self.stage(
-            denoiser=SimpleDenoiser(video_context, audio_context),
+            denoiser=simple_denoising_func(
+                video_context=video_context,
+                audio_context=audio_context,
+                transformer=self.model_ledger.transformer(),
+            ),
             sigmas=stage_1_sigmas,
             noiser=noiser,
             width=stage_1_output_shape.width,
@@ -250,7 +253,11 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
         )
 
         video_state, audio_state = self.stage(
-            denoiser=SimpleDenoiser(video_context, audio_context),
+            denoiser=simple_denoising_func(
+                video_context=video_context,
+                audio_context=audio_context,
+                transformer=self.model_ledger.transformer(),
+            ),
             sigmas=stage_2_sigmas,
             noiser=noiser,
             width=stage_2_output_shape.width,