vaishnav commited on
Commit
31a4877
·
1 Parent(s): 4651a01

revert embedding model

Browse files
Files changed (1) hide show
  1. 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 = "models/text-embedding-004"
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