Add support for ComfyUI

This commit is contained in:
lslowmotion 2024-05-02 19:05:57 +07:00
parent e32b2b2544
commit 034c98b50f
3 changed files with 15 additions and 13 deletions

View file

@ -3,14 +3,14 @@ FROM rocm/pytorch:rocm6.1_ubuntu22.04_py3.10_pytorch_2.1.2
ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
RUN apt-get update && apt-get install -y git && apt-get clean
RUN apt-get update && apt-get install -y git cargo nasm && apt-get clean
ENV ROOT=/stable-diffusion
RUN --mount=type=cache,target=/root/.cache/pip,Z \
git clone https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
git clone --recursive https://github.com/comfyanonymous/ComfyUI.git ${ROOT} && \
cd ${ROOT} && \
git checkout master && \
git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
# git reset --hard d1f3637a5a944d0607b899babd8ff11d87100503 && \
pip install -r requirements.txt
WORKDIR ${ROOT}