From 42502c9d8117273d11e3bf72eba626f733e8c20d Mon Sep 17 00:00:00 2001
From: ramblingcoder <markmcneelyjr@ramblingcoder.com>
Date: Tue, 4 Jul 2023 01:23:49 -0500
Subject: [PATCH] Use deploy resources instead of runtime nvidia

---
 docker/docker-compose.yml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 50a19662109..c58cf8bf397 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -1,6 +1,12 @@
 services:
   stable-diffusion-webui:
-    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
     image: emsi/stable-diffusion-webui
     build:
       context: ../