From e3d3071b1e7b4f125189f5cb4dbc8a3c209fdbaf Mon Sep 17 00:00:00 2001
From: yggdrasil75 <cblackburn7557@gmail.com>
Date: Sun, 10 Sep 2023 09:45:40 -0400
Subject: [PATCH] Update docker-compose.yml

fixes nvidia on windows
---
 docker/docker-compose.yml | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index c29be6b..6331532 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -1,7 +1,16 @@
 version: "3"
 services:
   sillytavern-extras:
-    runtime: nvidia
+#    runtime: nvidia
+     deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+#              either specify below which, or above how many. not both.
+#              device_ids: ['1'] 
+              capabilities: [gpu]
     image: cohee1207/sillytavern-extras
     build:
       context: ../