From 136fc43f23215197f527325b600a52392fecec6c Mon Sep 17 00:00:00 2001 From: Massaki Archambault Date: Thu, 4 Apr 2024 17:59:10 -0400 Subject: [PATCH] set OLLAMA_HOST --- docker-compose.amd.yml | 2 ++ docker-compose.cpu.yml | 2 ++ docker-compose.nvidia.yml | 2 ++ 3 files changed, 6 insertions(+) diff --git a/docker-compose.amd.yml b/docker-compose.amd.yml index 6279c97..f5325b7 100644 --- a/docker-compose.amd.yml +++ b/docker-compose.amd.yml @@ -9,6 +9,8 @@ services: entrypoint: /bootstrap.sh command: mistral network_mode: service:open-webui + environment: + OLLAMA_HOST: http://localhost:11434 # begin for AMD GPU support devices: - /dev/kfd diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml index 2b68fe2..362ff28 100644 --- a/docker-compose.cpu.yml +++ b/docker-compose.cpu.yml @@ -9,6 +9,8 @@ services: entrypoint: /bootstrap.sh command: mistral network_mode: service:open-webui + environment: + OLLAMA_HOST: http://localhost:11434 volumes: - ./ollama/bootstrap.sh:/bootstrap.sh:ro - ./ollama:/root/.ollama \ No newline at end of file diff --git a/docker-compose.nvidia.yml b/docker-compose.nvidia.yml index 5c7a18e..81c413b 100644 --- a/docker-compose.nvidia.yml +++ b/docker-compose.nvidia.yml @@ -9,6 +9,8 @@ services: entrypoint: /bootstrap.sh command: mistral network_mode: service:open-webui + environment: + OLLAMA_HOST: http://localhost:11434 # begin for NVIDIA GPU support deploy: resources: