| # 使用支持 CUDA 的基础镜像（如果要在 CPU 运行，基础镜像也通用） | |
| FROM python:3.10-slim | |
| # 设置工作目录 | |
| WORKDIR /app | |
| # 安装系统依赖 | |
| RUN apt-get update && apt-get install -y \ | |
| git \ | |
| libgl1-mesa-dri \ | |
| libglib2.0-0 \ | |
| wget \ | |
| && rm -rf /var/lib/apt/lists/* | |
| #libgl1-mesa-glx \ | |
| # 克隆 ComfyUI 源码 | |
| RUN git clone https://github.com/comfyanonymous/ComfyUI.git . | |
| # 安装依赖 (优先安装 Torch，确保版本匹配) | |
| RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| # 创建模型目录（可选：在此处预下载一个小模型，如 SD1.5 Pruned） | |
| #RUN wget -O models/checkpoints/v1-5-pruned-emaonly.safetensors \ | |
| # https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors | |
| #RUN git clone https://github.com/ltdrdata/ComfyUI-Manager custom_nodes/comfyui-manager | |
| # 暴露 Hugging Face 指定端口 | |
| EXPOSE 7860 | |
| ADD entrypoint.sh /app/entrypoint.sh | |
| # 启动命令：使用 --listen 监听所有接口，--port 7860 适配 HF | |
| # 如果是纯 CPU 环境，建议加上 --cpu 参数 | |
| #CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--cpu"] | |
| ENTRYPOINT ["sh", "/app/entrypoint.sh"] | |