docs: added comments

This commit is contained in:
Tymec
2023-04-14 14:53:18 +02:00
parent 64db4eef39
commit b042376db4
4 changed files with 7 additions and 0 deletions

View File

@@ -3,6 +3,7 @@ import abc
from config import AbstractSingleton, Config
import openai
# try to import sentence transformers, if it fails, default to ada
try:
from sentence_transformers import SentenceTransformer
except ImportError:
@@ -18,6 +19,7 @@ cfg = Config()
def get_embedding(text):
text = text.replace("\n", " ")
# use the embeder specified in the config
if cfg.memory_embeder == "sbert":
embedding = SentenceTransformer("sentence-transformers/all-mpnet-base-v2", device="cpu").encode(text, show_progress_bar=False)
else: