hugging2021 commited on
Commit
3c9ce69
·
verified ·
1 Parent(s): 36daa1c

Update concat_vector_store.py

Browse files
Files changed (1) hide show
  1. concat_vector_store.py +1 -1
concat_vector_store.py CHANGED
@@ -10,7 +10,7 @@ NEW_FOLDER = "new_documents" # Folder containing the new documents
10
  VECTOR_STORE_PATH = "vector_db"
11
 
12
  # 1. Loading the embedding model
13
- def get_embeddings(model_name="intfloat/multilingual-e5-large-instruct", device="cuda"):
14
  return E5Embeddings(
15
  model_name=model_name,
16
  model_kwargs={'device': device},
 
10
  VECTOR_STORE_PATH = "vector_db"
11
 
12
  # 1. Loading the embedding model
13
+ def get_embeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", device="cuda"):
14
  return E5Embeddings(
15
  model_name=model_name,
16
  model_kwargs={'device': device},