mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-11 14:02:37 +08:00
91494226db
- Remove model downloads from Dockerfile - Add symlink to /runpod-volume/models - Add extra_model_paths.yaml for network storage - Improve error handling in handler - Models will be loaded from RunPod network storage
75 lines
1.6 KiB
Docker
75 lines
1.6 KiB
Docker
# RunPod Serverless ComfyUI Worker
|
|
FROM runpod/pytorch:2.2.0-py3.11-cuda12.1.1-devel-ubuntu22.04
|
|
|
|
# Set working directory
|
|
WORKDIR /workspace
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y \
|
|
git \
|
|
wget \
|
|
curl \
|
|
unzip \
|
|
ffmpeg \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender-dev \
|
|
libglib2.0-0 \
|
|
libgl1-mesa-glx \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Copy ComfyUI source code
|
|
COPY . /workspace/ComfyUI
|
|
|
|
# Set ComfyUI as working directory
|
|
WORKDIR /workspace/ComfyUI
|
|
|
|
# Install Python dependencies
|
|
RUN pip install --no-cache-dir \
|
|
runpod \
|
|
requests \
|
|
pillow \
|
|
numpy \
|
|
torch \
|
|
torchvision \
|
|
torchaudio \
|
|
xformers \
|
|
accelerate \
|
|
transformers \
|
|
diffusers \
|
|
opencv-python \
|
|
scipy \
|
|
scikit-image \
|
|
safetensors \
|
|
bitsandbytes \
|
|
optimum
|
|
|
|
# Install ComfyUI requirements if exists
|
|
RUN if [ -f requirements.txt ]; then pip install --no-cache-dir -r requirements.txt; fi
|
|
|
|
# Create necessary directories for temp files
|
|
RUN mkdir -p /workspace/ComfyUI/input \
|
|
/workspace/ComfyUI/output \
|
|
/tmp/inputs \
|
|
/tmp/outputs \
|
|
/tmp/comfyui
|
|
|
|
# Create symlinks to network storage for models
|
|
RUN mkdir -p /runpod-volume && \
|
|
ln -sf /runpod-volume/models /workspace/ComfyUI/models
|
|
|
|
# Set environment variables
|
|
ENV PYTHONPATH="/workspace/ComfyUI:${PYTHONPATH}"
|
|
ENV COMFYUI_SERVERLESS=true
|
|
ENV NVIDIA_VISIBLE_DEVICES=all
|
|
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
|
|
|
# Set environment variables
|
|
ENV PYTHONPATH="/workspace/ComfyUI:${PYTHONPATH}"
|
|
ENV COMFYUI_SERVERLESS=true
|
|
|
|
# Expose port
|
|
EXPOSE 8000
|
|
|
|
# Start command
|
|
CMD ["python3", "runpod_handler.py"] |