Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ 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",
|
18 |
device_map="auto",
|
19 |
torch_dtype=torch.bfloat16
|
20 |
)
|
|
|
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",
|
19 |
torch_dtype=torch.bfloat16
|
20 |
)
|