From dcac9e487409a5ab795b95c97669109e2d5a8c8b Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Fri, 18 Oct 2024 11:12:27 -0700
Subject: [PATCH] update compose file

---
 .../distribution/docker/ollama/compose.yaml        | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/llama_stack/distribution/docker/ollama/compose.yaml b/llama_stack/distribution/docker/ollama/compose.yaml
index 22f5ded0e..2e3f85e45 100644
--- a/llama_stack/distribution/docker/ollama/compose.yaml
+++ b/llama_stack/distribution/docker/ollama/compose.yaml
@@ -26,7 +26,8 @@ services:
             capabilities: [gpu]
     runtime: nvidia
   llamastack-local-cpu:
-    depends_on: ollama
+    depends_on:
+    - ollama
     image: llamastack/llamastack-local-cpu
     network_mode: "host"
     volumes:
@@ -37,10 +38,11 @@ services:
       - "5000:5000"
     # Hack: wait for ollama server to start before starting docker
     entrypoint: bash -c "sleep 60; python -m llama_stack.distribution.server.server --yaml_config /root/llamastack-run-ollama.yaml"
-    restart_policy:
-      condition: on-failure
-      delay: 3s
-      max_attempts: 5
-      window: 60s
+    deploy:
+      restart_policy:
+        condition: on-failure
+        delay: 3s
+        max_attempts: 5
+        window: 60s
 volumes:
   ollama: