DeepMount00
commited on
Commit
•
28c6af2
1
Parent(s):
d1021ff
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
14 |
import torch
|
15 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
16 |
|
17 |
-
MODEL_NAME = "DeepMount00/Mistral-
|
18 |
|
19 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.bfloat16).eval()
|
20 |
model.to(device)
|
|
|
14 |
import torch
|
15 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
16 |
|
17 |
+
MODEL_NAME = "DeepMount00/Mistral-RAG"
|
18 |
|
19 |
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.bfloat16).eval()
|
20 |
model.to(device)
|