feat: 修改dcu的依赖

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=kolors-faceid
 ARG IMAGE_UPPER=Kolors-FaceID
-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 apt-get update && apt-get install -y gcc g++
-RUN pip install -r requirements.txt && \
-    pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
+RUN pip install -r requirements.txt
+RUN wget --no-check-certificate  --content-disposition \
+    'https://cancon.hpccube.com:65024/directlink/4/onnxruntime/DAS1.2/onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl' && \
+    pip install onnxruntime-1.15.0+das.opt1.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl \
+RUN pip install numpy==1.24.4
 
 #########
 # 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=kolors-faceid
 ARG IMAGE_UPPER=Kolors-FaceID
-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/Kwai-Kolors/Kolors /root/$IMAGE_UPPER/Kwai-Kolors/Kolors
 COPY chenyh/$IMAGE/Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus /root/$IMAGE_UPPER/Kwai-Kolors/Kolors-IP-Adapter-FaceID-Plus
 COPY chenyh/$IMAGE/antelopev2 /root/$IMAGE_UPPER/.insightface/models/antelopev2
-RUN apt-get update && apt install -y libgl1-mesa-glx libglib2.0-dev
+RUN apt-get update && apt install -y libglib2.0-dev
 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/$IMAGE/assets /root/
\ No newline at end of file

Kolors-FaceID/requirements.txt

 accelerate
 diffusers
 invisible_watermark
-torch
-transformers
-sentencepiece
-torchvision
 timm
-einops
 scipy
 insightface
-opencv-python
-gradio==3.40.0
+gradio
\ No newline at end of file
\ No newline at end of file