From 4b8672ac3b498799dc7bcd40ccc241945c2e86f7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 23 Mar 2025 02:39:34 +0200
Subject: [PATCH] Update model.py

---
 wanvideo/modules/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py
index cfa3c1c..bc3c242 100644
--- a/wanvideo/modules/model.py
+++ b/wanvideo/modules/model.py
@@ -32,7 +32,7 @@ def rope_riflex(pos, dim, theta, L_test, k, temporal):
     omega = 1.0 / (theta**scale)
 
     # RIFLEX modification - adjust last frequency component if L_test and k are provided
-    if temporal and k and L_test:
+    if temporal and k > 0 and L_test:
         omega[k-1] = 0.9 * 2 * torch.pi / L_test
 
     out = torch.einsum("...n,d->...nd", pos.to(dtype=torch.float32, device=device), omega)