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/vae22.py
View File

@@ -847,7 +847,7 @@ class Wan22VAEDecoder(nn.Module):
Returns:
video: [B, T', H', W', 3] decoded RGB in [-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()