klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: embedding function config (#175)

Browse Source
This commit is contained in:
cachho
2023-07-07 11:06:43 +02:00
committed by GitHub
parent 41a1dbab0e
commit 0c2653ccf5
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
embedchain/embedchain.py
View File

@@ -364,15 +364,14 @@ class OpenSourceApp(EmbedChain):
:param config: InitConfig instance to load as configuration. Optional. `ef` defaults to open source. :param config: InitConfig instance to load as configuration. Optional. `ef` defaults to open source.
""" """
print("Loading open source embedding model. This may take some time...") print("Loading open source embedding model. This may take some time...")
if not config or not config.ef: if not config:
if config is None: config = InitConfig(
config = InitConfig( ef = embedding_functions.SentenceTransformerEmbeddingFunction(
ef = embedding_functions.SentenceTransformerEmbeddingFunction( model_name="all-MiniLM-L6-v2"
model_name="all-MiniLM-L6-v2"
)
) )
else: )
config._set_embedding_function( elif not config.ef:
config._set_embedding_function(
embedding_functions.SentenceTransformerEmbeddingFunction( embedding_functions.SentenceTransformerEmbeddingFunction(
model_name="all-MiniLM-L6-v2" model_name="all-MiniLM-L6-v2"
)) ))