From 3a6712e3a002ac4ab246f0096c5c81ac95498a02 Mon Sep 17 00:00:00 2001 From: Yuhong Sun Date: Wed, 24 Jan 2024 12:00:25 -0800 Subject: [PATCH] Default Embedding Size (#993) --- backend/danswer/configs/model_configs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/danswer/configs/model_configs.py b/backend/danswer/configs/model_configs.py index e9c5029719e7..0edfa1d304e7 100644 --- a/backend/danswer/configs/model_configs.py +++ b/backend/danswer/configs/model_configs.py @@ -14,7 +14,7 @@ DOCUMENT_ENCODER_MODEL = ( os.environ.get("DOCUMENT_ENCODER_MODEL") or "thenlper/gte-small" ) # If the below is changed, Vespa deployment must also be changed -DOC_EMBEDDING_DIM = int(os.environ.get("DOC_EMBEDDING_DIM") or 0) +DOC_EMBEDDING_DIM = int(os.environ.get("DOC_EMBEDDING_DIM") or 384) # Model should be chosen with 512 context size, ideally don't change this DOC_EMBEDDING_CONTEXT_SIZE = 512 NORMALIZE_EMBEDDINGS = (