clean up

kyegomez · Jul 7, 2023 · eaa5d90 · eaa5d90
1 parent ed730c1
commit eaa5d90
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/LongNet/__init__.py b/LongNet/__init__.py
@@ -1,3 +1,4 @@
 from LongNet.attention import DilatedAttention
 from LongNet.model import LongNet, LongNetLanguage, LongNetSelector, Decoder, DecoderConfig
-from LongNet.training import Train
+from LongNet.training import Train
+from LongNet.model_test import LongNetTest
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 setup(
  name = 'LongNet',
  packages = find_packages(exclude=[]),
- version = '0.0.3',
+ version = '0.0.4',
  license='MIT',
  description = 'LongNet - Pytorch',
  author = 'Kye Gomez',