Update app.py

app.py

@@ -49,7 +49,6 @@ from ltx_core.components.noisers import GaussianNoiser
 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
-from ltx_core.model.video_vae.video_vae import decode_video as vae_decode_video
 from ltx_core.quantization import QuantizationPolicy
 from ltx_core.types import Audio, AudioLatentShape, VideoPixelShape
 from ltx_pipelines.distilled import DistilledPipeline
@@ -245,11 +244,10 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
         del video_encoder
         cleanup_memory()
 
-        decoded_video = vae_decode_video(
+        decoded_video = self.model_ledger.video_decoder().tiled_decode(
             video_state.latent,
-            self.model_ledger.video_decoder(),
             tiling_config,
-            generator,
+            generator=generator,
         )
         original_audio = Audio(
             waveform=decoded_audio.waveform.squeeze(0),