Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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
|
| 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 =
|
| 248 |
video_state.latent,
|
| 249 |
-
|
| 250 |
-
|
| 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),
|