Update app.py

app.py

@@ -48,7 +48,7 @@ from safetensors import safe_open
 import json
 import requests
 
-from ltx_core.components.diffusion_steps import Res2sDiffusionStep
+from ltx_core.components.diffusion_steps import Res2sDiffusionStep, EulerDiffusionStep
 from ltx_core.components.guiders import MultiModalGuider, MultiModalGuiderParams
 from ltx_core.components.noisers import GaussianNoiser
 from ltx_core.components.protocols import DiffusionStepProtocol
@@ -62,6 +62,7 @@ from ltx_core.types import Audio, AudioLatentShape, VideoLatentShape, VideoPixel
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_pipelines.utils import (
     res2s_audio_video_denoising_loop,
+    euler_denoising_loop
     cleanup_memory,
     encode_prompts,
     denoise_audio_video,
@@ -238,6 +239,7 @@ class LTX23DistilledA2VPipeline:
         generator = torch.Generator(device=self.device).manual_seed(seed)
         noiser = GaussianNoiser(generator=generator)
         stepper = Res2sDiffusionStep()
+        stage_2_stepper = EulerDiffusionStep()
         dtype = torch.bfloat16
 
         ctx_p, ctx_n = encode_prompts(
@@ -311,11 +313,11 @@ class LTX23DistilledA2VPipeline:
             )
 
         def stage2_denoising_loop(sigmas: torch.Tensor, video_state, audio_state, stepper: DiffusionStepProtocol):
-            return res2s_audio_video_denoising_loop(
+            return euler_denoising_loop(
                 sigmas=sigmas,
                 video_state=video_state,
                 audio_state=audio_state,
-                stepper=stepper,
+                stepper=stage_2_stepper,
                 denoise_fn=simple_denoising_func(
                     video_context=v_context_p,
                     audio_context=a_context_p,