chore: update to use ollama in docker-compose

This commit is contained in:
Javier Martinez 2024-07-24 09:14:41 +02:00
parent e657e67db0
commit fdd4ed6420
No known key found for this signature in database
2 changed files with 2 additions and 1 deletions

View file

@ -10,6 +10,7 @@ services:
PORT: 8080 PORT: 8080
PGPT_PROFILES: docker PGPT_PROFILES: docker
PGPT_MODE: ollama PGPT_MODE: ollama
PGPT_EMBED_MODE: ollama
ollama: ollama:
image: ollama/ollama:latest image: ollama/ollama:latest
volumes: volumes:

View file

@ -6,7 +6,7 @@ llm:
mode: ${PGPT_MODE:mock} mode: ${PGPT_MODE:mock}
embedding: embedding:
mode: ${PGPT_EMBEDDING_MODE:sagemaker} mode: ${PGPT_EMBED_MODE:mock}
llamacpp: llamacpp:
llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF} llm_hf_repo_id: ${PGPT_HF_REPO_ID:TheBloke/Mistral-7B-Instruct-v0.1-GGUF}