clean up

kyegomez · Jul 10, 2023 · ecd7de8 · ecd7de8
1 parent 0f762d7
commit ecd7de8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name="zetascale",
-    version="0.2.0",
+    version="0.0.2",
     author="Zeta Team",
     author_email="[email protected]",
     description="Transformers at zeta scales",

diff --git a/zeta/__init__.py b/zeta/__init__.py
@@ -7,4 +7,8 @@
 
 #models
 from zeta.models.KosmosX import KosmosTokenizer, Kosmos
-from zeta.models.LongNet import LongNetTokenizer, LongNet
+from zeta.models.LongNet import LongNetTokenizer, LongNet
+
+
+#attention
+from zeta.utils.attention.dilated_attention import DilatedAttention2