AbdBarho/stable-diffusion-webui-docker
1
0
Fork 0
mirror of https://github.com/AbdBarho/stable-diffusion-webui-docker.git synced 2025-12-06 07:12:11 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

add onnx

Browse source
This commit is contained in:
Brian Gebel 2025-04-29 07:22:34 -07:00
parent 024d49ba75
commit ce64863be9
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
services/comfy/Dockerfile
View file

@ -3,8 +3,9 @@ FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-runtime AS base
ENV DEBIAN_FRONTEND=noninteractive
ENV PIP_PREFER_BINARY=1
RUN apt-get update && \
apt-get install -y git wget curl libgl1-mesa-glx libglib2.0-0 && \
RUN --mount=type=cache,target=/var/cache/apt \
apt-get update && \
apt-get install -y git wget curl jq ffmpeg libgl1-mesa-glx libglib2.0-0 && \
apt-get clean
# Explore how we can use the CLI to do this instead
@ -25,10 +26,12 @@ WORKDIR /home/comfy/app
# conda create -n comfy-env python=3.11 && \
# conda activate comfy-env && \
# comfy install
RUN git clone --branch v0.3.30 --depth 1 https://github.com/comfyanonymous/ComfyUI.git . && \
RUN --mount=type=cache,target=/comfy/.cache/pip \
git clone --branch v0.3.30 --depth 1 https://github.com/comfyanonymous/ComfyUI.git . && \
mkdir -p .local/bin && \
pip install -r requirements.txt && \
# pip install onnxruntime-gpu # Should make it so this can be included maybe through a mountable requirements.txt or yaml?
pip install onnxruntime-gpu
# Should make ONNX optional by moving this into a configuration file that can be alterated at run time
COPY --chown=comfy:comfy . .