Add ComfyUIMini mobile-friendly UI integration
New features: - ComfyUIMini container (Node.js Alpine, ~150MB) for mobile/tablet access - Separate container architecture with shared Docker network - Health checks on both services with proper dependency ordering - Shared output volume for image gallery feature Files added: - comfyuimini/Dockerfile - Node.js 20 Alpine with tsx runtime - comfyuimini/.dockerignore - Build context filtering Files updated: - docker-compose.yml - Added comfyuimini service, network, health checks - .env.example - Added COMFYUIMINI_PORT and COMFYUIMINI_REF - README.md - Architecture diagram, ComfyUIMini docs, updated credits Access points: - ComfyUI (Desktop): http://<host>:8188 - ComfyUIMini (Mobile): http://<host>:3000 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -20,3 +20,8 @@ COMFYUI_FLAGS=--listen 0.0.0.0 --port 8188 --disable-pinned-memory --force-fp16
|
|||||||
# Build refs (pin to specific commits/tags for reproducibility)
|
# Build refs (pin to specific commits/tags for reproducibility)
|
||||||
COMFYUI_REF=master
|
COMFYUI_REF=master
|
||||||
SAGEATTN_REF=main
|
SAGEATTN_REF=main
|
||||||
|
|
||||||
|
# ComfyUIMini - Mobile-friendly UI (optional)
|
||||||
|
# Access at http://<host>:3000
|
||||||
|
COMFYUIMINI_PORT=3000
|
||||||
|
COMFYUIMINI_REF=main
|
||||||
|
|||||||
@@ -25,6 +25,7 @@ Standard ComfyUI containers and PyTorch wheels don't support sm_121. SparkyUI so
|
|||||||
|
|
||||||
- **ComfyUI** (latest master branch)
|
- **ComfyUI** (latest master branch)
|
||||||
- **ComfyUI-Manager** - auto-installed on first run for easy custom node management
|
- **ComfyUI-Manager** - auto-installed on first run for easy custom node management
|
||||||
|
- **ComfyUIMini** - mobile-friendly web UI for phones/tablets (separate container)
|
||||||
- **SageAttention** - compiled natively for sm_121 (Blackwell tensor cores)
|
- **SageAttention** - compiled natively for sm_121 (Blackwell tensor cores)
|
||||||
- **PyTorch 2.9.1+cu130** - ARM64 wheels with CUDA 13.0 support
|
- **PyTorch 2.9.1+cu130** - ARM64 wheels with CUDA 13.0 support
|
||||||
|
|
||||||
@@ -73,7 +74,9 @@ docker compose up -d
|
|||||||
docker compose logs -f
|
docker compose logs -f
|
||||||
```
|
```
|
||||||
|
|
||||||
**Access:** http://localhost:8188 (or your DGX Spark's IP on LAN)
|
**Access:**
|
||||||
|
- **ComfyUI (Desktop):** http://localhost:8188
|
||||||
|
- **ComfyUIMini (Mobile):** http://localhost:3000
|
||||||
|
|
||||||
## Requirements
|
## Requirements
|
||||||
|
|
||||||
@@ -102,24 +105,26 @@ SAGEATTN_REF=main
|
|||||||
## Architecture
|
## Architecture
|
||||||
|
|
||||||
```
|
```
|
||||||
┌─────────────────────────────────────────────────────────────┐
|
┌──────────────────────────────────────────────────────────────────┐
|
||||||
│ DGX Spark Host │
|
│ DGX Spark Host │
|
||||||
│ Ubuntu 24.04 (DGX OS 7) / Driver 580.x │
|
│ Ubuntu 24.04 (DGX OS 7) / Driver 580.x │
|
||||||
│ │
|
│ │
|
||||||
│ ┌─────────────────────────────────────────────────────┐ │
|
│ ┌────────────────────────────────────────────────────────────┐ │
|
||||||
│ │ Docker Container (sparkyui:cu130) │ │
|
│ │ Docker Network (sparky_net) │ │
|
||||||
│ │ │ │
|
│ │ │ │
|
||||||
│ │ CUDA 13.0.2 + PyTorch 2.9.1+cu130 │ │
|
│ │ ┌─────────────────────────┐ ┌──────────────────────────┐ │ │
|
||||||
│ │ SageAttention 2.2.0 (compiled for sm_121) │ │
|
│ │ │ comfyui (sparkyui:cu130)│ │ comfyuimini (node:20) │ │ │
|
||||||
│ │ ComfyUI 0.7.x + ComfyUI-Manager │ │
|
│ │ │ │ │ │ │ │
|
||||||
│ │ │ │
|
│ │ │ CUDA 13.0.2 + PyTorch │◄─┤ Mobile-friendly UI │ │ │
|
||||||
│ │ Key env vars: │ │
|
│ │ │ SageAttention (sm_121) │ │ REST + WebSocket proxy │ │ │
|
||||||
│ │ TORCH_CUDA_ARCH_LIST="12.1" │ │
|
│ │ │ ComfyUI + Manager │ │ │ │ │
|
||||||
│ │ TORCHDYNAMO_DISABLE="1" │ │
|
│ │ │ │ │ Shares /output volume │ │ │
|
||||||
│ └─────────────────────────────────────────────────────┘ │
|
│ │ └───────────┬─────────────┘ └────────────┬─────────────┘ │ │
|
||||||
│ │ │
|
│ │ │ │ │ │
|
||||||
│ Port 8188 (LAN) │
|
│ └──────────────┼─────────────────────────────┼────────────────┘ │
|
||||||
└─────────────────────────────────────────────────────────────┘
|
│ │ │ │
|
||||||
|
│ Port 8188 (Desktop) Port 3000 (Mobile) │
|
||||||
|
└──────────────────────────────────────────────────────────────────┘
|
||||||
```
|
```
|
||||||
|
|
||||||
## Version Compatibility
|
## Version Compatibility
|
||||||
@@ -184,6 +189,31 @@ nvidia-smi --query-gpu=clocks.sm,clocks.max.sm,persistence_mode --format=csv
|
|||||||
|
|
||||||
**Note:** GPU clock settings don't persist across reboots due to GB10 firmware behavior. Re-apply after each boot.
|
**Note:** GPU clock settings don't persist across reboots due to GB10 firmware behavior. Re-apply after each boot.
|
||||||
|
|
||||||
|
## ComfyUIMini (Mobile UI)
|
||||||
|
|
||||||
|
SparkyUI includes [ComfyUIMini](https://github.com/ImDarkTom/ComfyUIMini) - a lightweight, mobile-friendly web UI that runs in a separate container.
|
||||||
|
|
||||||
|
**Features:**
|
||||||
|
- Responsive design optimized for phones and tablets
|
||||||
|
- Simplified workflow execution interface
|
||||||
|
- Built-in image gallery (reads from shared output directory)
|
||||||
|
- Import workflows from ComfyUI in "API Format"
|
||||||
|
- Multiple themes (dark, light, aurora, nord, etc.)
|
||||||
|
|
||||||
|
**How it works:**
|
||||||
|
- Runs as a Node.js Express server in its own container (~150MB)
|
||||||
|
- Connects to ComfyUI via internal Docker network (`http://comfyui:8188`)
|
||||||
|
- Proxies REST API calls and WebSocket connections
|
||||||
|
- Shares the output directory for gallery viewing
|
||||||
|
|
||||||
|
**Access:** `http://<your-dgx-ip>:3000`
|
||||||
|
|
||||||
|
**Build only ComfyUIMini** (if ComfyUI already built):
|
||||||
|
```bash
|
||||||
|
docker compose build comfyuimini
|
||||||
|
docker compose up -d comfyuimini
|
||||||
|
```
|
||||||
|
|
||||||
## SageAttention Notes
|
## SageAttention Notes
|
||||||
|
|
||||||
SageAttention PR #297 added sm_121 support but was merged then reverted due to stability issues. Our approach:
|
SageAttention PR #297 added sm_121 support but was merged then reverted due to stability issues. Our approach:
|
||||||
@@ -204,8 +234,9 @@ When these land, SparkyUI can be simplified:
|
|||||||
## Credits
|
## Credits
|
||||||
|
|
||||||
- Unified memory architecture insights from [HurbaLurba's DGX-SPARK-COMFYUI-DOCKER](https://github.com/HurbaLurba/DGX-SPARK-COMFYUI-DOCKER)
|
- Unified memory architecture insights from [HurbaLurba's DGX-SPARK-COMFYUI-DOCKER](https://github.com/HurbaLurba/DGX-SPARK-COMFYUI-DOCKER)
|
||||||
- SageAttention by [thu-ml](https://github.com/thu-ml/SageAttention)
|
- [ComfyUIMini](https://github.com/ImDarkTom/ComfyUIMini) by ImDarkTom
|
||||||
- ComfyUI by [comfyanonymous](https://github.com/comfyanonymous/ComfyUI)
|
- [SageAttention](https://github.com/thu-ml/SageAttention) by thu-ml
|
||||||
|
- [ComfyUI](https://github.com/comfyanonymous/ComfyUI) by comfyanonymous
|
||||||
|
|
||||||
## License
|
## License
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,21 @@
|
|||||||
|
# Node.js
|
||||||
|
node_modules/
|
||||||
|
npm-debug.log*
|
||||||
|
|
||||||
|
# Git
|
||||||
|
.git/
|
||||||
|
.gitignore
|
||||||
|
|
||||||
|
# Documentation
|
||||||
|
*.md
|
||||||
|
LICENSE
|
||||||
|
|
||||||
|
# Environment
|
||||||
|
.env*
|
||||||
|
|
||||||
|
# IDE
|
||||||
|
.vscode/
|
||||||
|
.idea/
|
||||||
|
|
||||||
|
# Build artifacts (will be generated in container)
|
||||||
|
dist/
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
# ComfyUIMini - Mobile-friendly UI for ComfyUI
|
||||||
|
# Lightweight Node.js container that proxies to ComfyUI
|
||||||
|
FROM node:20-alpine
|
||||||
|
|
||||||
|
LABEL maintainer="SparkyUI"
|
||||||
|
LABEL description="ComfyUIMini mobile UI for ComfyUI on DGX Spark"
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# Install git for cloning, then clean up
|
||||||
|
RUN apk add --no-cache git
|
||||||
|
|
||||||
|
# Clone ComfyUIMini
|
||||||
|
ARG COMFYUIMINI_REF=main
|
||||||
|
RUN git clone https://github.com/ImDarkTom/ComfyUIMini.git . && \
|
||||||
|
git checkout ${COMFYUIMINI_REF}
|
||||||
|
|
||||||
|
# Install all deps, build TypeScript, then prune dev dependencies
|
||||||
|
RUN npm ci && \
|
||||||
|
npm run build && \
|
||||||
|
npm prune --omit=dev
|
||||||
|
|
||||||
|
# Create config directory and default config
|
||||||
|
# Config is overridden at runtime via NODE_CONFIG env var
|
||||||
|
RUN mkdir -p config && \
|
||||||
|
cp config/default.example.json config/default.json
|
||||||
|
|
||||||
|
# Create directories for workflows and ensure proper permissions
|
||||||
|
RUN mkdir -p workflows && \
|
||||||
|
chown -R node:node /app
|
||||||
|
|
||||||
|
# Switch to non-root user
|
||||||
|
USER node
|
||||||
|
|
||||||
|
EXPOSE 3000
|
||||||
|
|
||||||
|
# Health check - verify the server responds
|
||||||
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=10s --retries=3 \
|
||||||
|
CMD wget --no-verbose --tries=1 --spider http://localhost:3000/ || exit 1
|
||||||
|
|
||||||
|
# Project uses tsx to run TypeScript directly (not compiled to dist/)
|
||||||
|
CMD ["npm", "start"]
|
||||||
@@ -63,4 +63,63 @@ services:
|
|||||||
# Wheel cache (optional - for prebuilt wheels)
|
# Wheel cache (optional - for prebuilt wheels)
|
||||||
- ${SPARKYUI_DATA_PATH}/wheels:/opt/wheels
|
- ${SPARKYUI_DATA_PATH}/wheels:/opt/wheels
|
||||||
|
|
||||||
|
networks:
|
||||||
|
- sparky_net
|
||||||
|
|
||||||
|
# Health check - ComfyUI takes time to load, so generous start period
|
||||||
|
healthcheck:
|
||||||
|
test: ["CMD", "curl", "-f", "http://localhost:8188/"]
|
||||||
|
interval: 30s
|
||||||
|
timeout: 10s
|
||||||
|
start_period: 120s
|
||||||
|
retries: 3
|
||||||
|
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
|
||||||
|
# ComfyUIMini - Mobile-friendly UI
|
||||||
|
# Access at http://<host>:3000
|
||||||
|
comfyuimini:
|
||||||
|
build:
|
||||||
|
context: ./comfyuimini
|
||||||
|
dockerfile: Dockerfile
|
||||||
|
args:
|
||||||
|
COMFYUIMINI_REF: "${COMFYUIMINI_REF:-main}"
|
||||||
|
|
||||||
|
image: comfyuimini:latest
|
||||||
|
container_name: comfyuimini
|
||||||
|
|
||||||
|
ports:
|
||||||
|
- "${COMFYUIMINI_PORT:-3000}:3000"
|
||||||
|
|
||||||
|
environment:
|
||||||
|
# node-config override - connects to comfyui container via docker network
|
||||||
|
NODE_CONFIG: >-
|
||||||
|
{
|
||||||
|
"app_port": 3000,
|
||||||
|
"comfyui_url": "http://comfyui:8188",
|
||||||
|
"comfyui_ws_url": "ws://comfyui:8188",
|
||||||
|
"output_dir": "/shared/output",
|
||||||
|
"reject_unauthorised_cert": false
|
||||||
|
}
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
# Share output directory with ComfyUI for gallery feature (read-only)
|
||||||
|
- ${SPARKYUI_DATA_PATH}/output:/shared/output:ro
|
||||||
|
# Persist server-side workflows
|
||||||
|
- comfyuimini_workflows:/app/workflows
|
||||||
|
|
||||||
|
networks:
|
||||||
|
- sparky_net
|
||||||
|
|
||||||
|
depends_on:
|
||||||
|
comfyui:
|
||||||
|
condition: service_healthy
|
||||||
|
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
networks:
|
||||||
|
sparky_net:
|
||||||
|
driver: bridge
|
||||||
|
|
||||||
|
volumes:
|
||||||
|
comfyuimini_workflows:
|
||||||
|
|||||||
Reference in New Issue
Block a user