Update README.md
Browse filesedit model name and add "python" into script.
README.md
CHANGED
@@ -25,11 +25,11 @@ The dataset used is [OpenAssistant/oasst_top1_2023-08-25](https://huggingface.co
|
|
25 |
#### How to use
|
26 |
You will need the transformers>=4.31
|
27 |
Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
|
28 |
-
```
|
29 |
from transformers import AutoTokenizer
|
30 |
import transformers
|
31 |
import torch
|
32 |
-
model = "
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
34 |
pipeline = transformers.pipeline(
|
35 |
"text-generation",
|
|
|
25 |
#### How to use
|
26 |
You will need the transformers>=4.31
|
27 |
Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
|
28 |
+
```python
|
29 |
from transformers import AutoTokenizer
|
30 |
import transformers
|
31 |
import torch
|
32 |
+
model = "TinyLlama/TinyLlama-1.1B-Chat-v0.3"
|
33 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
34 |
pipeline = transformers.pipeline(
|
35 |
"text-generation",
|