feat: dcu版初始提交

CogVideoX-5B-Space/requirements.txt

 safetensors>=0.4.4
 spandrel>=0.3.4
 tqdm>=4.66.5
-opencv-python>=4.10.0.84
 scikit-video>=1.1.11
 diffusers>=0.30.1
 transformers>=4.44.0
 accelerate>=0.33.0
 sentencepiece>=0.2.0
 SwissArmyTransformer>=0.4.12
-numpy==1.26.0
-torch
-torchvision
 gradio>=4.42.0
 streamlit>=1.37.1
 imageio==2.34.2

Dockerfile

-FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04 as base
+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=gpu
+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
@@ -10,11 +10,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 #########
 # Prod  #
 #########
-FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-python3.10-cuda12.1-ubuntu22.04
+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.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/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