Refactor Wan model structure by renaming and relocating model imports from model.py to wan2.py, enhancing code organization and clarity across the Wan2 module.

2026-03-18 17:57:29 +01:00
parent 6c63163671
commit b029668cd2
13 changed files with 45 additions and 45 deletions
--- a/mlx_video/models/wan2/utils.py
+++ b/mlx_video/models/wan2/utils.py
@@ -21,7 +21,7 @@ def load_wan_model(
                      If provided, creates QuantizedLinear stubs before loading.
        loras: Optional list of (lora_path, strength) tuples to apply.
    """
-    from mlx_video.models.wan2.model import WanModel
+    from mlx_video.models.wan2.wan2 import WanModel

    model = WanModel(config)