We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 41d30ce + 342e19b commit 26283fbCopy full SHA for 26283fb
README.md
@@ -75,7 +75,7 @@ A simpler version is at [modules/mamba2_simple.py](mamba_ssm/modules/mamba2_simp
75
The usage is similar to Mamba(-1):
76
``` python
77
from mamba_ssm import Mamba2
78
-model = Mamba(
+model = Mamba2(
79
# This module uses roughly 3 * expand * d_model^2 parameters
80
d_model=dim, # Model dimension d_model
81
d_state=64, # SSM state expansion factor, typically 64 or 128
0 commit comments