Update unsloth.yaml

unsloth.yaml

@@ -1,23 +1,28 @@
 services:
   unsloth-train:
-    image: rocm/pytorch:latest
+    image: unsloth/unsloth-rocm   # pin a tag or digest once it works
     container_name: unsloth-train
     restart: unless-stopped
+    ipc: host
     devices:
       - /dev/kfd:/dev/kfd
       - /dev/dri/renderD128:/dev/dri/renderD128
     group_add:
-      - video
-    ipc: host
-    shm_size: 16g
+      - "44"    # getent group video | cut -d: -f3
+      - "992"   # getent group render | cut -d: -f3
+    ulimits:
+        memlock: -1
+        stack: 67108864
+    ports:
+        - 8075:8000
+        - 8777:8888
+    environment:
+        - JUPYTER_PASSWORD=mypassword
     volumes:
-      - unsloth-workspace:/workspace
-      - unsloth-cache:/root/.cache/huggingface
-    working_dir: /workspace
-    stdin_open: true
-    tty: true
-    # keeps the container running in the background so you can exec in, start a long training run, and detach without killing it. Use tmux or nohup inside the container for runs that should outlive your SSH session.
-    command: sleep infinity 
+        - /mnt/docker/server/unsloth:/workspace/host
+        - unsloth-cache:/workspace/.cache/huggingface
+        - unsloth-studio:/opt/unsloth-studio
+        
     labels:
       - homepage.group=Development
       - homepage.name=Unsloth
@@ -32,6 +37,6 @@ networks:
     external: true
 
 volumes:
-  unsloth-workspace:
   unsloth-cache:
+  unsloth-studio: