FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
ARG IMAGE=comfyui-photomaker-zho
ARG IMAGE_UPPER=ComfyUI-PhotoMaker-ZHO
ARG BRANCH=gpu
RUN cd /root && \
    git clone http://developer.hpccube.com/codes/chenpangpang/ComfyUI.git
RUN cd /root/ComfyUI && \
    pip install -r requirements.txt
RUN cd /root && \
    git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
RUN mv /root/$IMAGE/$IMAGE_UPPER /root/ComfyUI/custom_nodes/$IMAGE_UPPER && \
    cd /root/ComfyUI/custom_nodes/$IMAGE_UPPER && \
    pip install -r requirements.txt


#########
# Prod  #
#########
FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
ARG IMAGE=comfyui-photomaker-zho
ARG IMAGE_UPPER=ComfyUI-PhotoMaker-ZHO
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.2 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.2
#COPY chenyh/$IMAGE/coqui/XTTS-v2 /root/.local/share/tts/tts_models--multilingual--multi-dataset--xtts_v2
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/ComfyUI /root/ComfyUI
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh /root/