Add embedding mode config (#1541)

This commit is contained in:
Pablo Orgaz 2024-01-25 10:55:32 +01:00 committed by GitHub
parent 869233f0e4
commit 3e67e21d38
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions

View File

@ -5,6 +5,9 @@ server:
llm:
mode: ${PGPT_MODE:mock}
embedding:
mode: ${PGPT_MODE:sagemaker}
local:
llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF}
llm_hf_model_file: ${PGPT_HF_MODEL_FILE:mistral-7b-instruct-v0.1.Q4_K_M.gguf}
@ -16,4 +19,4 @@ sagemaker:
ui:
enabled: true
path: /
path: /