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/convert.py
+++ b/mlx_video/models/wan2/convert.py
@@ -594,7 +594,7 @@ def _quantize_saved_model(

    import mlx.nn as nn

-    from mlx_video.models.wan2.model import WanModel
+    from mlx_video.models.wan2.wan2 import WanModel

    if source_dir is None:
        source_dir = output_dir