BUG FIX: ImportError: cannot import name Transformer from partially i…

…nitialized module zeta.nn.architecture.transformer (most likely due to a circular import) (/usr/local/lib/python3.10/dist-packages/zeta/nn/architecture/transformer.py)
kyegomez · Sep 6, 2023 · abb9e17 · abb9e17
1 parent 6cba30e
commit abb9e17
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zetascale"
-version = "0.4.9"
+version = "0.5.1"
 description = "Transformers at zeta scales"
 authors = ["Zeta Team <[email protected]>"]
 license = "MIT"

diff --git a/zeta/nn/__init__.py b/zeta/nn/__init__.py
@@ -3,7 +3,7 @@
 
 
 # attention
-from zeta.nn.architecture.transformer import Transformer
+# from zeta.nn.architecture.transformer import Transformer
 from zeta.nn.architecture.local_transformer import LocalTransformer
 from zeta.nn.architecture.parallel_transformer import ParallelTransformerBlock
 from zeta.nn.architecture.auto_regressive_wrapper import AutoregressiveWrapper
@@ -22,15 +22,6 @@
 
 
 
-
-
-#utils
-from zeta.utils.main import *
-from zeta.utils.main import *
-from zeta.utils.main import *
-
-
-
 # embeddings
 from zeta.nn.embeddings.rope import RotaryEmbedding
 from zeta.nn.embeddings.xpos_relative_position import XPOS, rotate_every_two, apply_rotary_pos_emb