Update app.py

app.py

@@ -53,11 +53,13 @@ import requests
 
 from ltx_core.components.diffusion_steps import EulerDiffusionStep
 from ltx_core.components.noisers import GaussianNoiser
+from ltx_core.components.protocols import DiffusionStepProtocol
+from ltx_core.model.audio_vae import decode_audio as vae_decode_audio
 from ltx_core.model.audio_vae import encode_audio as vae_encode_audio
 from ltx_core.model.upsampler import upsample_video
 from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number, decode_video as vae_decode_video
 from ltx_core.quantization import QuantizationPolicy
-from ltx_core.types import Audio, AudioLatentShape, VideoPixelShape
+from ltx_core.types import Audio, LatentState, AudioLatentShape, VideoPixelShape
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_pipelines.utils import ModelLedger, euler_denoising_loop
 from ltx_pipelines.utils.args import ImageConditioningInput
@@ -66,6 +68,8 @@ from ltx_pipelines.utils.helpers import (
     cleanup_memory,
     combined_image_conditionings,
     denoise_video_only,
+    denoise_audio_video,
+    get_device,
     encode_prompts,
     simple_denoising_func,
 )