diff --git a/docker-compose.yml b/docker-compose.yml index 442fc5a..df84952 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -7,36 +7,20 @@ services: build: ./ # find docker file in designated path container_name: discord restart: always # rebuild container always - image: kevinthedang/discord-ollama:0.8.4 + image: kevinthedang/discord-ollama:latest environment: CLIENT_TOKEN: ${CLIENT_TOKEN} OLLAMA_IP: ${OLLAMA_IP} OLLAMA_PORT: ${OLLAMA_PORT} - MODEL: ${MODEL} REDIS_IP: ${REDIS_IP} REDIS_PORT: ${REDIS_PORT} + MODEL: ${MODEL} networks: ollama-net: ipv4_address: ${DISCORD_IP} volumes: - discord:/src/app # docker will not make this for you, make it yourself - # setup ollama container - ollama: - image: ollama/ollama:latest # build the image using ollama - container_name: ollama - restart: always - networks: - ollama-net: - ipv4_address: ${OLLAMA_IP} - runtime: nvidia # use Nvidia Container Toolkit for GPU support - devices: - - /dev/nvidia0 - volumes: - - ollama:/root/.ollama - ports: - - ${OLLAMA_PORT}:${OLLAMA_PORT} - # setup redis container redis: image: redis:latest @@ -60,6 +44,5 @@ networks: - subnet: ${SUBNET_ADDRESS}/16 volumes: - ollama: discord: redis: