Update README.md
Browse files
README.md
CHANGED
|
@@ -79,9 +79,9 @@ WIP
|
|
| 79 |
### Example of usage 👩💻
|
| 80 |
```py
|
| 81 |
import torch
|
| 82 |
-
from transformers import AutoModelForCausalLM, AutoTokenizer,
|
| 83 |
|
| 84 |
-
model_id = "mrm8488/
|
| 85 |
|
| 86 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 87 |
|
|
|
|
| 79 |
### Example of usage 👩💻
|
| 80 |
```py
|
| 81 |
import torch
|
| 82 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 83 |
|
| 84 |
+
model_id = "mrm8488/llama-2-coder-7b"
|
| 85 |
|
| 86 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 87 |
|