Spaces:
Sleeping
Sleeping
Update rag_engine.py
Browse files- rag_engine.py +2 -2
rag_engine.py
CHANGED
@@ -65,9 +65,9 @@ def load_model():
|
|
65 |
device = torch.device("cpu")
|
66 |
|
67 |
# Load tokenizer and model
|
68 |
-
tokenizer = AutoTokenizer.from_pretrained("intfloat/e5-
|
69 |
model = AutoModel.from_pretrained(
|
70 |
-
"intfloat/e5-
|
71 |
torch_dtype=torch.float16
|
72 |
)
|
73 |
|
|
|
65 |
device = torch.device("cpu")
|
66 |
|
67 |
# Load tokenizer and model
|
68 |
+
tokenizer = AutoTokenizer.from_pretrained("intfloat/e5-large-v2")
|
69 |
model = AutoModel.from_pretrained(
|
70 |
+
"intfloat/e5-large-v2",
|
71 |
torch_dtype=torch.float16
|
72 |
)
|
73 |
|