Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ HF-compatible version of Modello Italia 9B, developed by iGenius.
|
|
12 |
import torch
|
13 |
import transformers as tr
|
14 |
|
15 |
-
tokenizer = tr.AutoTokenizer.from_pretrained("riccorl/Modello-Italia-9B")
|
16 |
model = tr.AutoModelForCausalLM.from_pretrained(
|
17 |
"riccorl/Modello-Italia-9B-hf",
|
18 |
device_map="auto",
|
|
|
12 |
import torch
|
13 |
import transformers as tr
|
14 |
|
15 |
+
tokenizer = tr.AutoTokenizer.from_pretrained("riccorl/Modello-Italia-9B-hf")
|
16 |
model = tr.AutoModelForCausalLM.from_pretrained(
|
17 |
"riccorl/Modello-Italia-9B-hf",
|
18 |
device_map="auto",
|