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/init.py
+++ b/mlx_video/init.py
@@ -22,7 +22,7 @@ from mlx_video.models.ltx_2.utils import (
    load_safetensors,
    save_weights,
 )
-from mlx_video.models.wan import WanModel, WanModelConfig
+from mlx_video.models.wan2 import WanModel, WanModelConfig

 __all__ = [
    # Models