💬 Commit message: Update 2026-02-15 15:58:14, 1 files, 62 lines

📁 Files changed: 1
📝 Lines changed: 62

  • Dockerfile.rocm

Dockerfile.rocm

@@ -0,0 +1,62 @@
+# Tensors + ComfyUI for AMD ROCm (local testing)
+# tensors API exposed, ComfyUI internal only
+FROM saiden/sd-rocm
+
+WORKDIR /workspace
+
+# Install git and curl
+RUN apt-get update && apt-get install -y --no-install-recommends git curl && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install ComfyUI
+RUN git clone https://github.com/comfyanonymous/ComfyUI.git && \
+    cd ComfyUI && \
+    pip install --break-system-packages -r requirements.txt
+
+# Install ComfyUI Manager
+RUN cd /workspace/ComfyUI/custom_nodes && \
+    git clone https://github.com/ltdrdata/ComfyUI-Manager.git && \
+    cd ComfyUI-Manager && \
+    pip install --break-system-packages -r requirements.txt
+
+# Install tensors with server dependencies
+COPY . /tmp/tensors
+RUN pip install --break-system-packages /tmp/tensors'[server]' && \
+    rm -rf /tmp/tensors
+
+# Configure tensors
+RUN mkdir -p /root/.config/tensors && \
+    cat > /root/.config/tensors/config.toml << 'EOF'
+[paths]
+models_dir = "/workspace/ComfyUI/models"
+checkpoints = "/workspace/ComfyUI/models/checkpoints"
+loras = "/workspace/ComfyUI/models/loras"
+vae = "/workspace/ComfyUI/models/vae"
+embeddings = "/workspace/ComfyUI/models/embeddings"
+
+[comfyui]
+url = "http://127.0.0.1:8188"
+EOF
+
+# Startup script: ComfyUI internal + tensors API exposed
+RUN cat > /workspace/start.sh << 'EOF'
+#!/bin/bash
+# Start ComfyUI in background (internal only, localhost)
+python /workspace/ComfyUI/main.py --listen 127.0.0.1 --port 8188 &
+
+# Wait for ComfyUI to be ready
+echo "Waiting for ComfyUI..."
+until curl -s http://127.0.0.1:8188/system_stats > /dev/null 2>&1; do
+    sleep 1
+done
+echo "ComfyUI ready"
+
+# Start tensors API (exposed)
+exec tsr serve --host 0.0.0.0 --port 8080
+EOF
+RUN chmod +x /workspace/start.sh
+
+# Only expose tensors API
+EXPOSE 8080
+
+CMD ["/workspace/start.sh"]