diff --git a/unsloth.yaml b/unsloth.yaml
index c1b174d..15800e8 100644
--- a/unsloth.yaml
+++ b/unsloth.yaml
@@ -1,9 +1,9 @@
 services:
   unsloth-train:
-    image: unsloth/unsloth-rocm   # pin a tag or digest once it works
+    image: unsloth/unsloth-rocm:studio   # pin a tag or digest once it works
     container_name: unsloth-train
     restart: no 
-    command: sleep infinity
+    #command: sleep infinity do not use with studio version
     #The container runs as root by default. `--user <uid>:<gid>` is supported and keeps files on your mounts owned by you.
     ipc: host
     #deploy:
@@ -25,8 +25,8 @@ services:
         memlock: -1
         stack: 67108864
     ports:
-        - 8075:8000
-        - 8777:8888
+        - 8075:8000 # Unsloth Studio
+        - 8777:8888 # JupyterLab
     environment:
         - JUPYTER_PASSWORD=${JUPYTER_PWD}
         - UNSLOTH_STUDIO_PASSWORD=${UNSLOTH_STUDIO_PASSWORD:-}
@@ -38,9 +38,11 @@ services:
         - /srv/docker/server/unsloth/host:/workspace/host  # project directory
         - /srv/docker/server/unsloth/notebooks:/workspace/unsloth-notebooks # synced notebooks
         - unsloth-cache:/workspace/.cache/huggingface  # Model Downloads
-        - unsloth-data:/data #Unified persistence root: models, fine-tune outputs, exports, and auth db
+        #- unsloth-data:/data #Unified persistence root: models, fine-tune outputs, exports, and auth db
+        - unsloth-studio:/opt/unsloth-studio    # studio data
         - /mnt/raid1/data/hoard/medical_studies:/workspace/host/studies
         
+        
     labels:
       - homepage.group=Development
       - homepage.name=Unsloth
@@ -57,4 +59,5 @@ networks:
 volumes:
   unsloth-cache:
   unsloth-data:
+  unsloth-studio: