Refactor Wan model imports and update script paths in pyproject.toml; transition from wan to wan2 module structure for improved organization and clarity.

2026-03-18 17:52:30 +01:00
parent 17397da70c
commit 6c63163671
28 changed files with 354 additions and 1033 deletions
--- a/mlx_video/models/wan2/vae.py
+++ b/mlx_video/models/wan2/vae.py
@@ -589,7 +589,7 @@ class WanVAE(nn.Module):
        Returns:
            Video [B, 3, T_out, H_out, W_out] clamped to [-1, 1]
        """
-        from mlx_video.models.wan.tiling import TilingConfig, decode_with_tiling
+        from mlx_video.models.wan2.tiling import TilingConfig, decode_with_tiling

        if tiling_config is None:
            tiling_config = TilingConfig.default()