FROM docker.io/mixa3607/pytorch-gfx906:v2.9.0-rocm-7.0.2-20251102214219

WORKDIR /comfyui
COPY ./requirements.txt ./requirements.txt
RUN pip install -r requirements.txt
COPY ./ ./
ENTRYPOINT ["python", "main.py"]