|
FROM runpod/pytorch:2.2.1-py3.10-cuda12.1.1-devel-ubuntu22.04 |
|
WORKDIR /content |
|
ENV PATH="/home/camenduru/.local/bin:${PATH}" |
|
RUN adduser --disabled-password --gecos '' camenduru && \ |
|
adduser camenduru sudo && \ |
|
echo '%sudo ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers && \ |
|
chown -R camenduru:camenduru /content && \ |
|
chmod -R 777 /content && \ |
|
chown -R camenduru:camenduru /home && \ |
|
chmod -R 777 /home |
|
|
|
RUN apt update -y && add-apt-repository -y ppa:git-core/ppa && apt update -y && apt install -y aria2 git git-lfs unzip ffmpeg |
|
|
|
USER camenduru |
|
|
|
RUN pip install -q opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \ |
|
xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1 |
|
|
|
RUN git clone https://github.com/camenduru/ComfyUI /content/ComfyUI |
|
|
|
RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/TostModels/resolve/main/models/checkpoints/cyberrealisticXL_v11VAE.safetensors -d /content/ComfyUI/models/checkpoints -o model.safetensors |
|
|
|
COPY ./worker_runpod.py /content/ComfyUI/worker_runpod.py |
|
WORKDIR /content/ComfyUI |
|
CMD python worker_runpod.py |
|
|