Spaces:
Runtime error
Runtime error
Update concat_vector_store.py
Browse files- 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="
|
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},
|