From d152f441ff6a77354ef32909b2ec0b2c3a7bcc7d Mon Sep 17 00:00:00 2001 From: gcalgoz Date: Fri, 5 Jul 2024 18:14:58 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E5=AF=B9=E6=A8=A1=E5=9E=8B=E5=B5=8C?= =?UTF-8?q?=E5=85=A5=E5=BD=92=E4=B8=80=E5=8C=96,=E6=8F=90=E5=8D=87?= =?UTF-8?q?=E6=A3=80=E7=B4=A2=E9=80=9F=E5=BA=A6=20(#710)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/common/config/embedding_config.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/apps/common/config/embedding_config.py b/apps/common/config/embedding_config.py index fd5709d32..4d652bf95 100644 --- a/apps/common/config/embedding_config.py +++ b/apps/common/config/embedding_config.py @@ -24,10 +24,13 @@ class EmbeddingModel: model_name = CONFIG.get('EMBEDDING_MODEL_NAME') cache_folder = CONFIG.get('EMBEDDING_MODEL_PATH') device = CONFIG.get('EMBEDDING_DEVICE') + encode_kwargs = {'normalize_embeddings': True} e = HuggingFaceEmbeddings( model_name=model_name, cache_folder=cache_folder, - model_kwargs={'device': device}) + model_kwargs={'device': device}, + encode_kwargs=encode_kwargs, + ) EmbeddingModel.instance = e return EmbeddingModel.instance