Update app.py

app.py

@@ -55,10 +55,10 @@ from ltx_pipelines.distilled import DistilledPipeline
 from ltx_pipelines.utils import euler_denoising_loop
 from ltx_pipelines.utils.args import ImageConditioningInput
 from ltx_pipelines.utils.constants import DISTILLED_SIGMA_VALUES, STAGE_2_DISTILLED_SIGMA_VALUES
+from ltx_pipelines.pipeline_utils import denoise_audio_video
 from ltx_pipelines.utils.helpers import (
     cleanup_memory,
     combined_image_conditionings,
-    denoise_video_only,
     encode_prompts,
     simple_denoising_func,
 )
@@ -193,7 +193,7 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
             dtype=dtype,
             device=self.device,
         )
-        video_state = denoise_video_only(
+        video_state, _ = denoise_audio_video(
             output_shape=stage_1_output_shape,
             conditionings=stage_1_conditionings,
             noiser=noiser,
@@ -224,7 +224,7 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
             dtype=dtype,
             device=self.device,
         )
-        video_state = denoise_video_only(
+        video_state, _ = denoise_audio_video(
             output_shape=stage_2_output_shape,
             conditionings=stage_2_conditionings,
             noiser=noiser,