feat(wan): Add Wan2.1/2.2 T2V with quantization support

2026-02-26 16:16:07 +01:00
parent 7a74946c57
commit e64483a66a
21 changed files with 5309 additions and 35 deletions
--- a/mlx_video/init.py
+++ b/mlx_video/init.py
@@ -1,9 +1,12 @@
 from mlx_video.models.ltx import LTXModel, LTXModelConfig
+from mlx_video.models.wan import WanModel, WanModelConfig
 from mlx_video.convert import load_transformer_weights, load_vae_weights
 import os
 __all__ = [
    "LTXModel",
    "LTXModelConfig",
+    "WanModel",
+    "WanModelConfig",
    "load_transformer_weights",
    "load_vae_weights",
 ]