Update README.md
Browse files
README.md
CHANGED
|
@@ -29,7 +29,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 29 |
### Model Sources [optional]
|
| 30 |
|
| 31 |
- **Demo [optional]:**
|
| 32 |
-
|
| 33 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
|
| 34 |
import torch
|
| 35 |
|
|
@@ -91,8 +91,7 @@ _ = model.generate(
|
|
| 91 |
max_new_tokens=600,
|
| 92 |
pad_token_id=tokenizer.eos_token_id
|
| 93 |
)
|
| 94 |
-
|
| 95 |
-
```
|
| 96 |
|
| 97 |
## Uses
|
| 98 |
|
|
|
|
| 29 |
### Model Sources [optional]
|
| 30 |
|
| 31 |
- **Demo [optional]:**
|
| 32 |
+
```
|
| 33 |
from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
|
| 34 |
import torch
|
| 35 |
|
|
|
|
| 91 |
max_new_tokens=600,
|
| 92 |
pad_token_id=tokenizer.eos_token_id
|
| 93 |
)
|
| 94 |
+
```
|
|
|
|
| 95 |
|
| 96 |
## Uses
|
| 97 |
|