__init__

kyegomez · Jul 2, 2023 · c828802 · c828802
1 parent cf154a3
commit c828802
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/Andromeda/__init__.py b/Andromeda/__init__.py
@@ -1,2 +1 @@
-from old.traingv2 import TrainAndromeda
-from old.build_dataset import built_dataset
+from Andromeda.model import Andromeda
diff --git a/Andromeda/model.py b/Andromeda/model.py
@@ -1,6 +1,6 @@
 from optimus_prime import TransformerWrapper, AutoregressiveWrapper, AndromedaEmbedding, Decoder
 
-model = TransformerWrapper(
+Andromeda = TransformerWrapper(
     num_tokens=64007,
     max_seq_len=8192,
     use_abs_pos_emb=False,
@@ -24,4 +24,4 @@
     )
 )
 
-model = AutoregressiveWrapper(model)
+Andromeda = AutoregressiveWrapper(Andromeda)
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 setup(
-  name = 'Andromeda',
+  name = 'Andromeda-llm',
   packages = find_packages(exclude=['examples']),
   version = '0.0.1',
   license='MIT',