attention export

Kye · Kye · commit e18c46917532 · 2023-07-10T13:39:34.000-04:00
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
+
