Transformers
PyTorch
pt-sk commited on
Commit
78b5875
·
verified ·
1 Parent(s): 103fc0f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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("state-spaces/mamba-2.8b-slimpj")
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
  ```