ARG BASE_IMAGE=docker.m.daocloud.io/vllm/vllm-openai:v0.27.1-aarch64
FROM ${BASE_IMAGE}

USER root
ARG PIP_INDEX_URL=https://mirrors.aliyun.com/pypi/simple
ENV PIP_INDEX_URL=${PIP_INDEX_URL} PIP_DEFAULT_TIMEOUT=60 PIP_RETRIES=3
COPY sources/ComfyUI-source.tar.gz /tmp/ComfyUI-source.tar.gz
COPY sources/ComfyUI-MiniMax-H3-Turbo-source.tar.gz /tmp/ComfyUI-MiniMax-H3-Turbo-source.tar.gz
RUN echo "e42f1265f60cad3e649a3422bda97a462fabbe989db799f6587bb22f781886ba  /tmp/ComfyUI-source.tar.gz" | sha256sum -c - \
    && echo "a1398cac5a703a37073e5f74f852243d99bb716cf82d1e10eaa3b7cedf290a39  /tmp/ComfyUI-MiniMax-H3-Turbo-source.tar.gz" | sha256sum -c - \
    && mkdir -p /ComfyUI/custom_nodes/ComfyUI-MiniMax-H3-Turbo \
    && tar -xzf /tmp/ComfyUI-source.tar.gz --strip-components=1 -C /ComfyUI \
    && tar -xzf /tmp/ComfyUI-MiniMax-H3-Turbo-source.tar.gz --strip-components=1 -C /ComfyUI/custom_nodes/ComfyUI-MiniMax-H3-Turbo \
    && python3 -m pip install --no-cache-dir -r /ComfyUI/requirements.txt huggingface-hub==1.29.0 \
    && rm -f /tmp/ComfyUI-source.tar.gz /tmp/ComfyUI-MiniMax-H3-Turbo-source.tar.gz
COPY extra_model_paths.yaml /opt/megacube/extra_model_paths.yaml
WORKDIR /ComfyUI
ENTRYPOINT []
CMD ["python3", "/ComfyUI/main.py", "--listen", "0.0.0.0", "--port", "8000", "--fast", "autotune", "--extra-model-paths-config", "/opt/megacube/extra_model_paths.yaml"]
