[CLEANUP]

kyegomez · Apr 4, 2024 · 3f2ae1a · 3f2ae1a
1 parent d0fb831
commit 3f2ae1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bit_mamba.py b/bit_mamba.py
@@ -4,7 +4,7 @@
 # Create a random tensor of shape (2, 10, 512)
 x = torch.randn(2, 10, 512)
 
-# Create an instance of the BitMamba model with input size 512 and output size 6
+# Create an instance of the BitMamba model with input size 512 and depth size 6
 model = BitMamba(512, 6)
 
 # Pass the input tensor through the model to get the output

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "bitnet"
-version = "0.2.3"
+version = "0.2.4"
 description = "bitnet - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]