From 61100a007536d7fe6d942b4a469ffa1cb2dee71c Mon Sep 17 00:00:00 2001
From: lucidrains <lucidrains@gmail.com>
Date: Thu, 14 Dec 2023 12:54:09 -0800
Subject: [PATCH] remove beartype for forward

---
 rotary_embedding_torch/rotary_embedding_torch.py | 3 +--
 setup.py                                         | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/rotary_embedding_torch/rotary_embedding_torch.py b/rotary_embedding_torch/rotary_embedding_torch.py
index c678d9a..87363ca 100644
--- a/rotary_embedding_torch/rotary_embedding_torch.py
+++ b/rotary_embedding_torch/rotary_embedding_torch.py
@@ -239,11 +239,10 @@ def get_axial_freqs(self, *dims):
         return torch.cat(all_freqs, dim = -1)
 
     @autocast(enabled = False)
-    @beartype
     def forward(
         self,
         t: Tensor,
-        seq_len: Optional[int] = None,
+        seq_len = None,
         offset = 0
     ):
         should_cache = (
diff --git a/setup.py b/setup.py
index ff7d654..a174222 100644
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'rotary-embedding-torch',
   packages = find_packages(),
-  version = '0.5.0',
+  version = '0.5.1',
   license='MIT',
   description = 'Rotary Embedding - Pytorch',
   long_description_content_type = 'text/markdown',