diff --git a/unsloth.yaml b/unsloth.yaml
index a55df63..37df900 100644
--- a/unsloth.yaml
+++ b/unsloth.yaml
@@ -2,7 +2,8 @@ services:
   unsloth-train:
     image: unsloth/unsloth-rocm   # pin a tag or digest once it works
     container_name: unsloth-train
-    restart: unless-stopped
+    restart: no 
+    command: sleep infinity
     #The container runs as root by default. `--user <uid>:<gid>` is supported and keeps files on your mounts owned by you.
     ipc: host
     #deploy: