[FEATS][__Init__]

kyegomez · Mar 4, 2024 · 620c435 · 620c435
1 parent 763d4e9
commit 620c435
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/example.py b/example.py
@@ -17,4 +17,4 @@
 # Forward pass
 y = model(x)
 
-print(y)
+print(y)
diff --git a/griffin_torch/__init__.py b/griffin_torch/__init__.py
@@ -1,3 +1,9 @@
-from griffin_torch.main import Griffin
+from griffin_torch.main import output_head, RMSNorm, GriffinResidualBlock, Griffin
 
-__all__ = ["Griffin"]
+
+__all__ = [
+    output_head,
+    RMSNorm,
+    GriffinResidualBlock,
+    Griffin,
+]
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "griffin-torch"
-version = "0.0.2"
+version = "0.0.3"
 description = "Griffin - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]