Init docker support

This commit is contained in:
LEv145 2023-03-21 14:54:16 +02:00 committed by GitHub
parent e391ffbfa9
commit 89a2f464d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,16 +1,19 @@
FROM python:3.10-slim
ENV CLI_ARGS=""
COPY . /opt/comfy_ui
WORKDIR /opt/comfy_ui
RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/cu117 \
torch==1.13.1 \
torch==1.13.1+cu117 \
torchvision \
torchaudio \
xformers
xformers \
triton
COPY . /opt/comfy_ui
RUN pip install -r requirements.txt
ENV CLI_ARGS=""
CMD python main.py ${CLI_ARGS}