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

refactor(wan): move causal_temporal tiling to wan/tiling.py

Browse Source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
Daniel
2026-03-11 12:02:54 +01:00
parent 1cf878f5e0
commit c144c8817c
4 changed files with 287 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx_video/models/wan/vae.py
View File

@@ -549,7 +549,7 @@ class WanVAE(nn.Module):
Returns:
Video [B, 3, T_out, H_out, W_out] clamped to [-1, 1]
"""
from mlx_video.models.ltx.video_vae.tiling import TilingConfig, decode_with_tiling
from mlx_video.models.wan.tiling import TilingConfig, decode_with_tiling
if tiling_config is None:
tiling_config = TilingConfig.default()