From 0f277beb05a1551d00a5793d72017246c553751f Mon Sep 17 00:00:00 2001
From: pixelthorn <forager@vivaldi.net>
Date: Wed, 28 Oct 2026 09:07:44 -0500
Subject: [PATCH] Update unsloth.yaml

---
 unsloth.yaml | 29 +++++++++++++++++------------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/unsloth.yaml b/unsloth.yaml
index 5c1688a..4421e16 100644
--- a/unsloth.yaml
+++ b/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: