FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 as base
RUN cd /root && git clone -b dcu http://developer.hpccube.com/codes/chenpangpang/instantid.git
WORKDIR /root/instantid/InstantID
COPY chenyh/InstantID/InstantID /root/instantid/InstantID/checkpoints
COPY chenyh/InstantID/antelopev2 /root/instantid/InstantID/models/antelopev2
COPY chenyh/InstantID/wangqixun /root/instantid/InstantID/wangqixun
RUN apt-get update && apt-get install -y gcc g++
RUN pip install -r gradio_demo/requirements.txt && \
    wget https://cancon.hpccube.com:65024/directlink/4/onnxruntime/DAS1.1/onnxruntime-1.15.0+das1.1.git739f24d.abi1.dtk2404-cp310-cp310-manylinux_2_31_x86_64.whl && \
    pip install onnxruntime-1.15.0+das1.1.git739f24d.abi1.dtk2404-cp310-cp310-manylinux_2_31_x86_64.whl

#########
# Prod  #
#########
FROM image.sourcefind.cn:5000/dcu/admin/base/jupyterlab-pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
RUN apt-get update && apt-get install -y libglib2.0-0 libgl1-mesa-glx
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/instantid/InstantID /root/InstantID
COPY --from=base /root/instantid/启动器.ipynb /root/instantid/run.sh /root/