remove adding staticmethod

lucidrains · Jul 6, 2024 · 72e0a27 · 72e0a27
1 parent 74cc6b0
commit 72e0a27
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/local_attention/rotary.py b/local_attention/rotary.py
@@ -30,8 +30,6 @@ def __init__(
         scale = (torch.arange(0, dim, 2) + 0.4 * dim) / (1.4 * dim)
         self.register_buffer('scale', scale, persistent = False)
 
-        self.apply_rotary_pos_emb = staticmethod(apply_rotary_pos_emb)
-
     @autocast(enabled = False)
     def forward(self, x):
         seq_len, device = x.shape[-2], x.device

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'local-attention',
   packages = find_packages(),
-  version = '1.9.10',
+  version = '1.9.11',
   license='MIT',
   description = 'Local attention, window with lookback, for language modeling',
   long_description_content_type = 'text/markdown',