feat: 更新依赖版本

Dockerfile

-FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.3.1-py3.10-cuda11.8-ubuntu22.04-devel as base
+FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda12.1-ubuntu22.04-devel  as base
 ARG IMAGE=nvcomposer
 ARG IMAGE_UPPER=NVComposer
 ARG BRANCH=gpu
@@ -10,7 +10,7 @@ RUN pip install -r requirements.txt
 #########
 # Prod  #
 #########
-FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.3.1-py3.10-cuda11.8-ubuntu22.04-devel
+FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.1.2-py3.10-cuda12.1-ubuntu22.04-devel
 ARG IMAGE=nvcomposer
 ARG IMAGE_UPPER=NVComposer
 COPY chenyh/$IMAGE/frpc_linux_amd64_* /opt/conda/lib/python3.10/site-packages/gradio/

NVComposer/requirements.txt

-pytorch_lightning==2.3
-deepspeed
-taming-transformers
-scipy
-einops
-kornia
-open_clip_torch
-openai-clip
-xformers
-timm
-av
-gradio
\ No newline at end of file
+pytorch_lightning==2.1.2
+deepspeed==0.15.0
+taming-transformers==0.0.1
+diffusers==0.30.1
+transformers
+torchvision==0.16.2
+scipy==1.14.1
+einops==0.8.0
+kornia==0.7.3
+open_clip_torch==2.20.0
+openai-clip==1.0.1
+numpy==1.26.3
+xformers==0.0.23.post1
+timm==1.0.9
+av==12.3.0
+gradio==5.8.0
\ No newline at end of file