FROM python:3.10-slim


WORKDIR /opt/comfy_ui


RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/rocm5.4.2 \
torch \
torchvision \
torchaudio

COPY ../.. /opt/comfy_ui
RUN pip install -r requirements.txt


ENV CLI_ARGS=""
CMD python main.py ${CLI_ARGS}