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/SG161222/RealVisXL_V3.0 /root/ComfyUI/SG161222/RealVisXL_V3.0
COPY chenyh/$IMAGE/TencentARC/PhotoMaker/photomaker-v1.bin /root/ComfyUI/photomaker-v1.bin
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/
COPY --from=base /root/$IMAGE/xiaozhan.jpeg /root/ComfyUI/input/