Chris4K commited on
Commit
8999d94
·
1 Parent(s): 5a2e3b2

Update vector_store_retriever.py

Browse files
Files changed (1) hide show
  1. vector_store_retriever.py +2 -3
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="hkunlp/instructor-large",
10
- embed_instruction="Represent the document for retrieval: ",
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