From 720b571fa77c730d668deedb631b491a6e575b9c Mon Sep 17 00:00:00 2001
From: pixelthorn <forager@vivaldi.net>
Date: Wed, 28 Oct 2026 08:43:13 -0500
Subject: [PATCH] Add unsloth.yaml

---
 unsloth.yaml | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 unsloth.yaml

diff --git a/unsloth.yaml b/unsloth.yaml
new file mode 100644
index 0000000..5c1688a
--- /dev/null
+++ b/unsloth.yaml
@@ -0,0 +1,37 @@
+services:
+  unsloth-train:
+    image: rocm/pytorch:latest
+    container_name: unsloth-train
+    restart: unless-stopped
+    devices:
+      - /dev/kfd:/dev/kfd
+      - /dev/dri/renderD128:/dev/dri/renderD128
+    group_add:
+      - video
+    ipc: host
+    shm_size: 16g
+    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 
+    labels:
+      - homepage.group=Development
+      - homepage.name=Unsloth
+      - homepage.icon=unsloth.png
+      - homepage.description=LLM Training
+      - rackpeek.tags=ai
+    networks:
+      - ollama-net
+
+networks:
+  ollama-net:
+    external: true
+
+volumes:
+  unsloth-workspace:
+  unsloth-cache:
+