Update imports

kyegomez · Apr 6, 2024 · d3941ee · d3941ee
1 parent cb58448
commit d3941ee
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/zeta/models/andromeda.py b/zeta/models/andromeda.py
@@ -1,7 +1,7 @@
 # the best llm ever made
 from torch.nn import Module
 
-from zeta.structs.auto_regressive_wrapper import AutoregressiveWrapper
+from zeta.structs.auto_regressive_wrapper import AutoRegressiveWrapper
 from zeta.structs.transformer import Decoder, Transformer
 
 
@@ -74,7 +74,7 @@ def __init__(
                 ),
             )
 
-            self.decoder = AutoregressiveWrapper(self.Andromeda)
+            self.decoder = AutoregRessiveWrapper(self.Andromeda)
 
         except Exception as e:
             print("Failed to initialize Andromeda: ", e)

diff --git a/zeta/models/gpt4.py b/zeta/models/gpt4.py
@@ -1,7 +1,7 @@
 import torch
 from torch import Tensor, nn
 
-from zeta.structs.auto_regressive_wrapper import AutoregressiveWrapper
+from zeta.structs.auto_regressive_wrapper import AutoRegressiveWrapper
 from zeta.structs.transformer import (
     Decoder,
     Encoder,
@@ -81,7 +81,7 @@ def __init__(
                 ),
             )
 
-            self.decoder = AutoregressiveWrapper(self.decoder)
+            self.decoder = AutoRegressiveWrapper(self.decoder)
 
         except Exception as e:
             print("Failed to initialize Andromeda: ", e)

diff --git a/zeta/models/llama.py b/zeta/models/llama.py
@@ -1,4 +1,4 @@
-from zeta.structs.auto_regressive_wrapper import AutoregressiveWrapper
+from zeta.structs.auto_regressive_wrapper import AutoRegressiveWrapper
 from zeta.structs.transformer import Decoder, Transformer
 
 
@@ -28,7 +28,7 @@ def __init__(
                 rotary_xpos=rotary_xpos,
             ),
         )
-        self.decoder = AutoregressiveWrapper(self.decoder)
+        self.decoder = AutoRegressiveWrapper(self.decoder)
 
     def forward(self, text):
         model_input = self.decoder.forward(text)[0]

diff --git a/zeta/models/palme.py b/zeta/models/palme.py
@@ -1,6 +1,6 @@
 import torch
 
-from zeta.structs.auto_regressive_wrapper import AutoregressiveWrapper
+from zeta.structs.auto_regressive_wrapper import AutoRegressiveWrapper
 from zeta.structs.transformer import (
     Decoder,
     Encoder,
@@ -57,7 +57,7 @@ def __init__(
             ),
         )
 
-        self.decoder = AutoregressiveWrapper(self.decoder)
+        self.decoder = AutoRegressiveWrapper(self.decoder)
 
     def forward(self, img, text):
         try:

diff --git a/zeta/structs/__init__.py b/zeta/structs/__init__.py
@@ -1,4 +1,4 @@
-from zeta.structs.auto_regressive_wrapper import AutoregressiveWrapper
+from zeta.structs.auto_regressive_wrapper import AutoRegressiveWrapper
 from zeta.structs.clip_encoder import CLIPVisionTower, build_vision_tower
 from zeta.structs.encoder_decoder import EncoderDecoder
 from zeta.structs.hierarchical_transformer import (
@@ -21,7 +21,7 @@
 from zeta.structs.transformer_block import TransformerBlock
 
 __all__ = [
-    "AutoregressiveWrapper",
+    "AutoRegressiveWrapper",
     "Encoder",
     "Decoder",
     "EncoderDecoder",