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, EulerDiffusionStep
+from ltx_core.components.diffusion_steps import EulerDiffusionStep
 from ltx_core.components.guiders import MultiModalGuider, MultiModalGuiderParams
 from ltx_core.components.noisers import GaussianNoiser
 from ltx_core.components.protocols import DiffusionStepProtocol
@@ -171,8 +171,7 @@ class LTX23DistilledA2VPipeline:
 
         generator = torch.Generator(device=self.device).manual_seed(seed)
         noiser = GaussianNoiser(generator=generator)
-        stepper = Res2sDiffusionStep()
-        euler_stepper = EulerDiffusionStep()
+        stepper = EulerDiffusionStep()
         dtype = torch.bfloat16
 
         ctx_p, ctx_n = encode_prompts(
@@ -250,7 +249,7 @@ class LTX23DistilledA2VPipeline:
                 sigmas=sigmas,
                 video_state=video_state,
                 audio_state=audio_state,
-                stepper=euler_stepper,
+                stepper=stepper,
                 denoise_fn=simple_denoising_func(
                     video_context=v_context_p,
                     audio_context=a_context_p,
@@ -316,7 +315,7 @@ class LTX23DistilledA2VPipeline:
             conditionings=stage_2_conditionings,
             noiser=noiser,
             sigmas=stage_2_sigmas,
-            stepper=euler_stepper,
+            stepper=stepper,
             denoising_loop_fn=stage2_denoising_loop,
             components=self.pipeline_components,
             dtype=dtype,