Hugging Face's logo

pt-sk
/
mamba_ml_abstract

Transformers
PyTorch
mamba
Test Generation
research abstract
Inference Endpoints
Model card Files Community
1
pt-sk commited on
Commit
06776d0
·
verified ·
1 Parent(s): 9e6566b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -19,10 +19,13 @@ Import the scripts from the code folder
19
  from model import Mamba, ModelArgs
20
  ```
21
 
22
- Loading Model```
 
 
23
  mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
24
  ```
25
 
 
26
  Loading Tokenizer
27
  ```
28
  tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')
 
19
  from model import Mamba, ModelArgs
20
  ```
21
 
22
+
23
+ Loading Model
24
+ ```
25
  mamba_model = Mamba.from_pretrained("pt-sk/mamba").to("cuda")
26
  ```
27
 
28
+
29
  Loading Tokenizer
30
  ```
31
  tokenizer = AutoTokenizer.from_pretrained('pt-sk/mamba')