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 |
|
|
|