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/tests/wan_test_helpers.py
+++ b/tests/wan_test_helpers.py
@@ -3,7 +3,7 @@

 def _make_tiny_config():
    """Create a tiny WanModelConfig for testing."""
-    from mlx_video.models.wan.config import WanModelConfig
+    from mlx_video.models.wan2.config import WanModelConfig

    config = WanModelConfig()
    # Override to tiny values