n0p/mlx-video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enhance video generation with progress bar for streaming and remove debug prints from tiling decoder

Browse Source
This commit is contained in:
Prince Canuma
2026-01-17 23:53:53 +01:00
parent f256c5fb25
commit b1bf9e2dc0
3 changed files with 5 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx_video/models/ltx/video_vae/decoder.py
View File

@@ -506,13 +506,8 @@ class LTX2VideoDecoder(nn.Module):
if not needs_spatial_tiling and not needs_temporal_tiling:
# No tiling needed, use regular decode
if debug:
print("[Tiling] Input fits within tile size, using regular decode")
return self(sample, causal=causal, timestep=timestep, debug=debug, chunked_conv=use_chunked_conv)
if debug:
print(f"[Tiling] Using tiled decode (spatial={needs_spatial_tiling}, temporal={needs_temporal_tiling})")
return decode_with_tiling(
decoder_fn=self,
latents=sample,
@@ -521,7 +516,6 @@ class LTX2VideoDecoder(nn.Module):
temporal_scale=8, # VAE temporal upsampling factor
causal=causal,
timestep=timestep,
debug=debug,
chunked_conv=use_chunked_conv,
on_frames_ready=on_frames_ready,
)