Update README.md
Browse files
README.md
CHANGED
@@ -39,8 +39,8 @@ import torch
|
|
39 |
|
40 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
41 |
|
42 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
43 |
-
model = AutoModelForCausalLM.from_pretrained("
|
44 |
model.to(device)
|
45 |
|
46 |
|
|
|
39 |
|
40 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
41 |
|
42 |
+
tokenizer = AutoTokenizer.from_pretrained("e-palmisano/Phi3-ITA-mini-4k-instruct", trust_remote_code=True)
|
43 |
+
model = AutoModelForCausalLM.from_pretrained("e-palmisano/Phi3-ITA-mini-4k-instruct", trust_remote_code=True)
|
44 |
model.to(device)
|
45 |
|
46 |
|