diff --git a/docker-compose-13b.yml b/docker-compose-13b.yml
index 39c7831..ce1c5fc 100644
--- a/docker-compose-13b.yml
+++ b/docker-compose-13b.yml
@@ -6,6 +6,7 @@ services:
     build:
       context: ./api
       dockerfile: 13B.Dockerfile
+    restart: on-failure
     environment:
       MODEL: '/models/llama-2-13b-chat.bin'
 
@@ -13,6 +14,7 @@ services:
     image: 'ghcr.io/getumbrel/llama-gpt-ui:latest'
     ports:
       - 3000:3000
+    restart: on-failure
     environment:
       - 'OPENAI_API_KEY=sk-XXXXXXXXXXXXXXXXXXXX'
       - 'OPENAI_API_HOST=http://llama-gpt-api:8000'
diff --git a/docker-compose-70b.yml b/docker-compose-70b.yml
index c4c433b..80d4aac 100644
--- a/docker-compose-70b.yml
+++ b/docker-compose-70b.yml
@@ -6,6 +6,7 @@ services:
     build:
       context: ./api
       dockerfile: 70B.Dockerfile
+    restart: on-failure
     environment:
       MODEL: '/models/llama-2-70b-chat.bin'
 
@@ -13,6 +14,7 @@ services:
     image: 'ghcr.io/getumbrel/llama-gpt-ui:latest'
     ports:
       - 3000:3000
+    restart: on-failure
     environment:
       - 'OPENAI_API_KEY=sk-XXXXXXXXXXXXXXXXXXXX'
       - 'OPENAI_API_HOST=http://llama-gpt-api:8000'
diff --git a/docker-compose.yml b/docker-compose.yml
index 155311e..4d9a9a4 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -6,13 +6,15 @@ services:
     build:
       context: ./api
       dockerfile: Dockerfile
+    restart: on-failure
     environment:
       MODEL: '/models/llama-2-7b-chat.bin'
 
   llama-gpt-ui:
-    image: ghcr.io/getumbrel/llama-gpt-ui
+    image: 'ghcr.io/getumbrel/llama-gpt-ui:latest'
     ports:
       - 3000:3000
+    restart: on-failure
     environment:
       - 'OPENAI_API_KEY=sk-XXXXXXXXXXXXXXXXXXXX'
       - 'OPENAI_API_HOST=http://llama-gpt-api:8000'