From 876155a2fee87c06a588eccc00b051f44320c428 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Tue, 16 Sep 2025 17:38:42 -0700
Subject: [PATCH] update docker compose

---
 docker-compose.yml | 20 ++++----------------
 1 file changed, 4 insertions(+), 16 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 6a82fbc..67e8dfb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,23 +1,17 @@
 services:
   ollama:
-    volumes:
-      - ./data/ollama/ollama:/root/.ollama
     container_name: ai-ollama
+    runtime: nvidia
     pull_policy: always
     tty: true
     restart: unless-stopped
     image: docker.io/ollama/ollama:latest
+    volumes:
+      - ./data/ollama/ollama:/root/.ollama
     ports:
       - 7869:11434
     environment:
       - OLLAMA_KEEP_ALIVE=24h
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: all
-              capabilities: [gpu]
 
   ollama-webui:
     image: ghcr.io/open-webui/open-webui:main
@@ -65,19 +59,13 @@ services:
   comfyui:
     container_name: ai-comfyui
     image: mistox/comfyui-ampere
+    runtime: nvidia
     volumes:
       - ./data/comfyui/models:/app/comfyui/models
       - ./data/comfyui/output:/app/comfyui/output
     ports:
       - 8010:8188
     restart: always
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: all
-              capabilities: [gpu]
 
 volumes:
   uvicorn-src: