Commit
•
02d8365
1
Parent(s):
7aacc36
Fix formatting
Browse files
README.md
CHANGED
@@ -68,6 +68,7 @@ input_ids = tokenizer("What is a mamba?", return_tensors="pt")["input_ids"]
|
|
68 |
|
69 |
out = model.generate(input_ids, max_new_tokens=10)
|
70 |
print(tokenizer.batch_decode(out))
|
|
|
71 |
|
72 |
### Model hyperparameters
|
73 |
|
@@ -87,5 +88,4 @@ More details about the model hyperparameters are given in the table below :
|
|
87 |
| state_size | 16 |
|
88 |
| dtype | bfloat16 |
|
89 |
| tie_word_embeddings | false |
|
90 |
-
| non embeddings params | 1.27B |
|
91 |
-
```
|
|
|
68 |
|
69 |
out = model.generate(input_ids, max_new_tokens=10)
|
70 |
print(tokenizer.batch_decode(out))
|
71 |
+
```
|
72 |
|
73 |
### Model hyperparameters
|
74 |
|
|
|
88 |
| state_size | 16 |
|
89 |
| dtype | bfloat16 |
|
90 |
| tie_word_embeddings | false |
|
91 |
+
| non embeddings params | 1.27B |
|
|