FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel as base
ARG IMAGE=cogvideox-5b
ARG IMAGE_UPPER=CogVideoX-5B-Space
ARG BRANCH=dcu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN pip install --no-cache-dir -r requirements.txt


#########
# Prod  #
#########
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel
ARG IMAGE=cogvideox-5b
ARG IMAGE_UPPER=CogVideoX-5B-Space
COPY chenyh/$IMAGE/frpc_linux_amd64_v0.3 /opt/conda/lib/python3.10/site-packages/gradio/
RUN chmod +x /opt/conda/lib/python3.10/site-packages/gradio/frpc_linux_amd64_v0.3
COPY chenyh/$IMAGE/model_real_esran /root/$IMAGE_UPPER/model_real_esran
COPY chenyh/$IMAGE/model_rife /root/$IMAGE_UPPER/model_rife
COPY chenyh/$IMAGE/THUDM/CogVideoX-5b /root/$IMAGE_UPPER/THUDM/CogVideoX-5b
RUN apt-get update && apt-get install -y libglib2.0-0 libgl1-mesa-glx build-essential && apt-get clean
COPY --from=base /opt/conda/lib/python3.10/site-packages /opt/conda/lib/python3.10/site-packages
COPY --from=base /root/$IMAGE/$IMAGE_UPPER /root/$IMAGE_UPPER
COPY --from=base /root/$IMAGE/启动器.ipynb /root/$IMAGE/start.sh  /root/
COPY --from=base /root/$IMAGE/assets /root/assets