attention export

kyegomez · Jul 10, 2023 · e18c469 · e18c469
1 parent ecd7de8
commit e18c469
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/zeta/__init__.py b/zeta/__init__.py
@@ -11,4 +11,6 @@
 
 
 #attention
-from zeta.utils.attention.dilated_attention import DilatedAttention2
+from zeta.utils.attention.dilated_attention import DilatedAttention2
+from zeta.utils.attention.main import Attention, AttentionLayers
+