Update README.md
Browse files
README.md
CHANGED
@@ -48,8 +48,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
48 |
|
49 |
device = "cuda"
|
50 |
|
51 |
-
model = AutoModelForCausalLM.from_pretrained("ilsp/Llama-Krikri-8B-
|
52 |
-
tokenizer = AutoTokenizer.from_pretrained("ilsp/Llama-Krikri-8B-
|
53 |
|
54 |
model.to(device)
|
55 |
|
|
|
48 |
|
49 |
device = "cuda"
|
50 |
|
51 |
+
model = AutoModelForCausalLM.from_pretrained("ilsp/Llama-Krikri-8B-Instruct")
|
52 |
+
tokenizer = AutoTokenizer.from_pretrained("ilsp/Llama-Krikri-8B-Instruct")
|
53 |
|
54 |
model.to(device)
|
55 |
|