From bcdfee27d32bd41eec147e0989e4ef0e3cde5bfb Mon Sep 17 00:00:00 2001
From: Julien Bongars <julienbongars23@gmail.com>
Date: Sun, 4 Feb 2024 19:17:20 +0800
Subject: [PATCH] chore: add failure policy for automatic1111

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

diff --git a/docker-compose.yml b/docker-compose.yml
index 44dcc87..2b0cb6a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,20 +1,22 @@
 version: '3.9'
 
 x-base_service: &base_service
-    ports:
-      - "${WEBUI_PORT:-7860}:7860"
-    volumes:
-      - &v1 ./data:/data
-      - &v2 ./output:/output
-    stop_signal: SIGKILL
-    tty: true
-    deploy:
-      resources:
-        reservations:
-          devices:
-              - driver: nvidia
-                device_ids: ['0']
-                capabilities: [compute, utility]
+  ports:
+    - "${WEBUI_PORT:-7860}:7860"
+  volumes:
+    - &v1 ./data:/data
+    - &v2 ./output:/output
+  stop_signal: SIGKILL
+  tty: true
+  deploy:
+    resources:
+      reservations:
+        devices:
+          - driver: nvidia
+            device_ids: ["0"]
+            capabilities: [compute, utility]
+    restart_policy:
+      condition: on-failure
 
 name: webui-docker
 
@@ -64,7 +66,6 @@ services:
     environment:
       - CLI_ARGS=
 
-
   comfy-cpu:
     <<: *comfy
     profiles: ["comfy-cpu"]