dagloop5 commited on
Commit
8fa2949
·
verified ·
1 Parent(s): a964a04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -48,7 +48,7 @@ from ltx_core.components.diffusion_steps import EulerDiffusionStep
48
  from ltx_core.components.noisers import GaussianNoiser
49
  from ltx_core.model.audio_vae import encode_audio as vae_encode_audio
50
  from ltx_core.model.upsampler import upsample_video
51
- from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number, decode_video as vae_decode_video
52
  from ltx_core.quantization import QuantizationPolicy
53
  from ltx_core.types import Audio, AudioLatentShape, VideoPixelShape
54
  from ltx_pipelines.distilled import DistilledPipeline
@@ -244,11 +244,10 @@ class LTX23DistilledA2VPipeline(DistilledPipeline):
244
  del video_encoder
245
  cleanup_memory()
246
 
247
- decoded_video = vae_decode_video(
248
  video_state.latent,
249
- self.model_ledger.video_decoder(),
250
- tiling_config,
251
- generator,
252
  )
253
  original_audio = Audio(
254
  waveform=decoded_audio.waveform.squeeze(0),
 
48
  from ltx_core.components.noisers import GaussianNoiser
49
  from ltx_core.model.audio_vae import encode_audio as vae_encode_audio
50
  from ltx_core.model.upsampler import upsample_video
51
+ from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number
52
  from ltx_core.quantization import QuantizationPolicy
53
  from ltx_core.types import Audio, AudioLatentShape, VideoPixelShape
54
  from ltx_pipelines.distilled import DistilledPipeline
 
244
  del video_encoder
245
  cleanup_memory()
246
 
247
+ decoded_video = _video_decoder.decode_video(
248
  video_state.latent,
249
+ tiling_config=tiling_config,
250
+ generator=generator,
 
251
  )
252
  original_audio = Audio(
253
  waveform=decoded_audio.waveform.squeeze(0),