Spaces:
Runtime error
Runtime error
vaishnav commited on
Commit ·
31a4877
1
Parent(s): 4651a01
revert embedding model
Browse files- stores/chroma.py +1 -1
stores/chroma.py
CHANGED
|
@@ -5,7 +5,7 @@ from langchain_chroma import Chroma
|
|
| 5 |
class ChromaDB:
|
| 6 |
def __init__(self, embeddings):
|
| 7 |
self._persistent_directory = "embeddings"
|
| 8 |
-
model_name = "
|
| 9 |
model_kwargs = {'device': 'cpu'}
|
| 10 |
encode_kwargs = {'normalize_embeddings': False}
|
| 11 |
self.embeddings = embeddings
|
|
|
|
| 5 |
class ChromaDB:
|
| 6 |
def __init__(self, embeddings):
|
| 7 |
self._persistent_directory = "embeddings"
|
| 8 |
+
model_name = "sentence-transformers/all-mpnet-base-v2"
|
| 9 |
model_kwargs = {'device': 'cpu'}
|
| 10 |
encode_kwargs = {'normalize_embeddings': False}
|
| 11 |
self.embeddings = embeddings
|