More poodles

2026-03-11 08:14:12 +01:00
parent d207275fea
commit 1cf878f5e0
9 changed files with 347 additions and 23 deletions
--- a/mlx_video/models/wan/model.py
+++ b/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,