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/test_wan_lora.py
+++ b/tests/test_wan_lora.py
@@ -331,7 +331,7 @@ class TestEndToEnd:
    """End-to-end LoRA loading and application."""

    def test_load_and_apply_loras(self):
-        from mlx_video.convert_wan import load_and_apply_loras
+        from mlx_video.models.wan2.convert import load_and_apply_loras

        with tempfile.TemporaryDirectory() as tmp:
            # Create mock LoRA safetensors