diff --git a/docker-compose.yml b/docker-compose.yml
index 41659d5..50dee88 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -107,10 +107,7 @@ services:
       - ${TMP_HOST_DIR:-./tmp}:/tmpdl
       - ${MODELS_HOST_DIR:-./models}:/root/.cache/huggingface
       - ./app:/app
-    device_requests:
-      - driver: nvidia
-        count: ${NVIDIA_DEVICE_COUNT:-1}
-        capabilities: [video, compute, utility]
+    runtime: ${DOCKER_GPU_RUNTIME:-runc}
     depends_on: [meili, redis, open-webui]
     restart: unless-stopped
     healthcheck:
@@ -173,10 +170,7 @@ services:
       - ${TMP_HOST_DIR:-./tmp}:/tmpdl
       - ${MODELS_HOST_DIR:-./models}:/root/.cache/huggingface
       - ./app:/app
-    device_requests:
-      - driver: nvidia
-        count: ${NVIDIA_DEVICE_COUNT:-1}
-        capabilities: [video, compute, utility]
+    runtime: ${DOCKER_GPU_RUNTIME:-runc}
     depends_on: [meili, redis, open-webui]
     restart: unless-stopped
     healthcheck: