Remove Wan2 model files, including configuration, attention mechanisms, and utility functions, to streamline the codebase and eliminate unused components. This cleanup enhances maintainability and focuses on the core functionality of the Wan2 module.

2026-03-18 17:59:43 +01:00
parent b029668cd2
commit 996a542011
37 changed files with 354 additions and 354 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.wan2 import WanModel, WanModelConfig
+from mlx_video.models.wan_2 import WanModel, WanModelConfig

 __all__ = [
    # Models