Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -6,9 +6,8 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
|
|
6 |
|
7 |
# Initialize the HuggingFaceInstructEmbeddings
|
8 |
hf = HuggingFaceInstructEmbeddings(
|
9 |
-
model_name="
|
10 |
-
|
11 |
-
query_instruction="Represent the query for retrieval: "
|
12 |
)
|
13 |
|
14 |
# Load and process the PDF files
|
|
|
6 |
|
7 |
# Initialize the HuggingFaceInstructEmbeddings
|
8 |
hf = HuggingFaceInstructEmbeddings(
|
9 |
+
model_name="sentence-transformers/all-MiniLM-L6-v2",
|
10 |
+
model_kwargs={"device": "cpu"}
|
|
|
11 |
)
|
12 |
|
13 |
# Load and process the PDF files
|