Spaces:
Runtime error
Runtime error
change init model to use cpu
Browse files
app.py
CHANGED
|
@@ -43,7 +43,7 @@ print("load embeddings")
|
|
| 43 |
embedding_model = HuggingFaceEmbeddings(
|
| 44 |
model_name="intfloat/e5-base-v2",
|
| 45 |
# multi_process=True,
|
| 46 |
-
model_kwargs={"device": "
|
| 47 |
encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
|
| 48 |
)
|
| 49 |
print("load vector store")
|
|
|
|
| 43 |
embedding_model = HuggingFaceEmbeddings(
|
| 44 |
model_name="intfloat/e5-base-v2",
|
| 45 |
# multi_process=True,
|
| 46 |
+
model_kwargs={"device": "cpu"}, # use cuda for faster embeddings on nbidia GPUs
|
| 47 |
encode_kwargs={"normalize_embeddings": True}, # Set `True` for cosine similarity
|
| 48 |
)
|
| 49 |
print("load vector store")
|