From cdcf1b0c7bc8bcffd8c78a8ea34bf4b91ea9a996 Mon Sep 17 00:00:00 2001 From: Quentin McGaw Date: Mon, 25 Mar 2024 11:28:56 +0000 Subject: [PATCH] `PGPT_EMBEDDING_MODE` to define embedding mode --- Dockerfile.external | 1 + Dockerfile.local | 1 + settings-docker.yaml | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Dockerfile.external b/Dockerfile.external index b618d482..38264146 100644 --- a/Dockerfile.external +++ b/Dockerfile.external @@ -24,6 +24,7 @@ ENV \ PYTHONPATH="$PYTHONPATH:/home/worker/app/private_gpt/" \ APP_ENV=prod \ PGPT_MODE=mock \ + PGPT_EMBEDDING_MODE=sagemaker \ PGPT_HF_REPO_ID=TheBloke/Mistral-7B-Instruct-v0.1-GGUF \ PGPT_HF_MODEL_FILE=mistral-7b-instruct-v0.1.Q4_K_M.gguf \ PGPT_EMBEDDING_HF_MODEL_NAME=BAAI/bge-small-en-v1.5 \ diff --git a/Dockerfile.local b/Dockerfile.local index 0c169051..d6d90f48 100644 --- a/Dockerfile.local +++ b/Dockerfile.local @@ -35,6 +35,7 @@ ENV \ PYTHONPATH="$PYTHONPATH:/home/worker/app/private_gpt/" \ APP_ENV=prod \ PGPT_MODE=mock \ + PGPT_EMBEDDING_MODE=sagemaker \ PGPT_HF_REPO_ID=TheBloke/Mistral-7B-Instruct-v0.1-GGUF \ PGPT_HF_MODEL_FILE=mistral-7b-instruct-v0.1.Q4_K_M.gguf \ PGPT_EMBEDDING_HF_MODEL_NAME=BAAI/bge-small-en-v1.5 \ diff --git a/settings-docker.yaml b/settings-docker.yaml index d8037fa6..8e218f19 100644 --- a/settings-docker.yaml +++ b/settings-docker.yaml @@ -6,7 +6,7 @@ llm: mode: ${PGPT_MODE:mock} embedding: - mode: ${PGPT_MODE:sagemaker} + mode: ${PGPT_EMBEDDING_MODE:sagemaker} llamacpp: llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF}