Skip to content

Commit 26283fb

Browse files
authored
Merge pull request #357 from YoniChechik/patch-1
Update README.md - typo fix: Mamba2 should be used
2 parents 41d30ce + 342e19b commit 26283fb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ A simpler version is at [modules/mamba2_simple.py](mamba_ssm/modules/mamba2_simp
7575
The usage is similar to Mamba(-1):
7676
``` python
7777
from mamba_ssm import Mamba2
78-
model = Mamba(
78+
model = Mamba2(
7979
# This module uses roughly 3 * expand * d_model^2 parameters
8080
d_model=dim, # Model dimension d_model
8181
d_state=64, # SSM state expansion factor, typically 64 or 128

0 commit comments

Comments
 (0)