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/vae22.py
+++ b/mlx_video/models/wan2/vae22.py
@@ -966,7 +966,7 @@ class Wan22VAEDecoder(nn.Module):
        Returns:
            video: [B, T', H', W', 3] decoded RGB in [-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()