Adam Ladachowski
63 lines
1.7 KiB
Docker
63 lines
1.7 KiB
Docker
# 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"]
|