diff --git a/requirements.txt b/requirements.txt
index 28a4ee0..f2eae51 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 diffusers
 einops
-huggingface_hub>=0.25.2
-ninja~=1.11.1.4
-transformers[timm]>=4.45.0
+huggingface_hub
+ninja
+transformers[timm]
diff --git a/tricks/modules/ltx_model.py b/tricks/modules/ltx_model.py
index 59b62d9..20e5f29 100644
--- a/tricks/modules/ltx_model.py
+++ b/tricks/modules/ltx_model.py
@@ -6,7 +6,7 @@
 from comfy.ldm.lightricks.model import (
     BasicTransformerBlock,
     LTXVModel,
-    apply_rotary_emb,
+    apply_rope1,
     precompute_freqs_cis,
 )
 from comfy.ldm.lightricks.symmetric_patchifier import latent_to_pixel_coords
@@ -57,8 +57,8 @@ def forward(self, x, context=None, mask=None, pe=None, transformer_options={}):
         k = self.k_norm(k)
 
         if pe is not None:
-            q = apply_rotary_emb(q, pe)
-            k = apply_rotary_emb(k, pe)
+            q = apply_rope1(q.unsqueeze(1), pe).squeeze(1)
+            k = apply_rope1(k.unsqueeze(1), pe).squeeze(1)
 
         feta_score = None
         if (