fastembed gpu
Browse files- backend/populate_vec_db_and_seach.py +1 -1
- requirements.txt +1 -1
backend/populate_vec_db_and_seach.py
CHANGED
|
@@ -196,5 +196,5 @@ def get_client() -> QdrantClient:
|
|
| 196 |
)
|
| 197 |
|
| 198 |
def get_model() -> TextEmbedding:
|
| 199 |
-
return TextEmbedding(model_name=EMBEDDING_MODEL_NAME)
|
| 200 |
|
|
|
|
| 196 |
)
|
| 197 |
|
| 198 |
def get_model() -> TextEmbedding:
|
| 199 |
+
return TextEmbedding(model_name=EMBEDDING_MODEL_NAME,providers=["CUDAExecutionProvider", "CPUExecutionProvider"])
|
| 200 |
|
requirements.txt
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
-
qdrant-client[fastembed]
|
| 2 |
openai
|
|
|
|
| 1 |
+
qdrant-client[fastembed-gpu]
|
| 2 |
openai
|