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