Update README.md
Browse files
README.md
CHANGED
@@ -8,5 +8,5 @@ Model code: https://github.com/state-spaces/mamba/tree/main
|
|
8 |
To load the model, follow the installation instruction in the code repo, and then:
|
9 |
```
|
10 |
from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
|
11 |
-
model = MambaLMHeadModel.from_pretrained("
|
12 |
```
|
|
|
8 |
To load the model, follow the installation instruction in the code repo, and then:
|
9 |
```
|
10 |
from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel
|
11 |
+
model = MambaLMHeadModel.from_pretrained("pt-sk/mamba-2.8b-slimpj")
|
12 |
```
|