More poodles

mlx_video/models/wan/README.md

@@ -146,12 +146,16 @@ For example, for using the the distilled [Wan2.2-Lightning](https://huggingface.
 python -m mlx_video.generate_wan \
     --model-dir /Volumes/SSD/Wan-AI/Wan2.2-T2V-A14B-MLX \
     --width 480 \
-    --height 480 \
-    --num-frames 121 \
-    --prompt "Two dogs of the poodle breed sitting on a beach wearing sunglasses, close up, cinematic, sunset" \
+    --height 704 \
+    --num-frames 41 \
+    --prompt "Two dogs of the poodle breed sitting on a beach wearing sunglasses, nodding with their heads, close up, cinematic, sunset" \
     --steps 4 \
     --guide-scale 1 \
     --trim-first-frames 1 \
+    --seed 2391784614 \
     --lora-high /Volumes/SSD/Wan-AI/lightx2v/Wan2.2-Lightning/Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V2.0/high_noise_model.safetensors 1 \
     --lora-low /Volumes/SSD/Wan-AI/lightx2v/Wan2.2-Lightning/Wan2.2-T2V-A14B-4steps-lora-rank64-Seko-V2.0/low_noise_model.safetensors 1
  ```
+
+Which results in 
+![Poodles](../../../examples/poodles-wan.gif)

mlx_video/models/wan/config.py

@@ -1,5 +1,5 @@
-from dataclasses import dataclass, field
-from typing import List, Optional, Tuple, Union
+from dataclasses import dataclass
+from typing import Tuple, Union
 
 from mlx_video.models.ltx.config import BaseModelConfig
 
@@ -104,7 +104,7 @@ class WanModelConfig(BaseModelConfig):
             sample_shift=5.0,
             sample_guide_scale=(3.5, 3.5),
             max_area=704 * 1280,
-
+        )
 
     @classmethod
     def wan22_ti2v_5b(cls) -> "WanModelConfig":
@@ -126,4 +126,4 @@ class WanModelConfig(BaseModelConfig):
             sample_guide_scale=5.0,
             sample_fps=24,
             max_area=704 * 1280,
-
+        )

mlx_video/models/wan/docs/DIAGNOSTICS.md

@@ -315,11 +315,6 @@ Applied alongside bug fixes to improve inference speed:
 - **Redundant type cast removal**: MLX type promotion handles `bfloat16 * float32 → float32` automatically — removed 240 unnecessary graph nodes per step (6 casts × 40 blocks)
 - **Euler scheduler sync fix**: Pre-store sigmas as Python floats to avoid `.item()` evaluation sync
 
-### TeaCache Integration
-- Polynomial rescaling stays in MLX lazy graph (Horner's method)
-- Single `.item()` call on the accumulated distance for the skip/compute decision
-- Configurable threshold, retention steps, and cutoff steps
-
 ---
 
 ## Resolved: CFG Effectiveness (was Open Investigation)

mlx_video/models/wan/model.py

@@ -1,5 +1,4 @@
 import math
-
 import mlx.core as mx
 import mlx.nn as nn
 import numpy as np
@@ -354,7 +353,6 @@ class WanModel(nn.Module):
             for i, sl in enumerate(seq_lens_list):
                 attn_mask[i, :, :, sl:] = -1e9
 
-
         kwargs = dict(
             e=e0,
             seq_lens=seq_lens_list,