From 655bef00f27045d4d129cf2194ff516731631d7c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8F=AD=E6=89=AC?= <xingjun.wxj@alibaba-inc.com>
Date: Sun, 28 Sep 2025 14:48:16 +0800
Subject: [PATCH] update deepspeed==0.17.4 due to the gpu image imports error

---
 docker/build_image.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docker/build_image.py b/docker/build_image.py
index 9d0fb0fa..1f3f0b81 100644
--- a/docker/build_image.py
+++ b/docker/build_image.py
@@ -209,6 +209,7 @@ class GPUImageBuilder(Builder):
 RUN pip install tf-keras==2.16.0 --no-dependencies && \
     pip install onnx==1.18.0 --no-dependencies && \
     pip install transformers==4.55.4 --no-dependencies && \
+    pip install deepspeed==0.17.4 --no-dependencies && \
     pip install --no-cache-dir torchsde jupyterlab torchmetrics==0.11.4 basicsr pynvml shortuuid && \
     CUDA_HOME=/usr/local/cuda TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6 8.9 9.0" \
         pip install --no-cache-dir  'git+https://github.com/facebookresearch/detectron2.git'