Update rope_embedding.py

2024-11-05 00:08:16 -08:00 · 2024-11-05 00:08:16 -08:00 · 4076dc49cb
commit 4076dc49cb
parent 0ab731d40b
1 changed files with 3 additions and 3 deletions
--- a/unsloth/kernels/rope_embedding.py
+++ b/unsloth/kernels/rope_embedding.py
@ -25,9 +25,9 @@ def _rope_embedding(
    cos, cos_row_stride,
    sin, sin_row_stride,
    seqlen,
-    head_dim      : tl.constexpr,
-    n_heads       : tl.constexpr,
-    BACKWARD_PASS : tl.constexpr,
+    head_dim,
+    n_heads,
+    BACKWARD_PASS,
    BLOCK_SIZE    : tl.constexpr,
 ):
    """