Update unsloth.yaml

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: