From 29cea7350e9526b46fa3a4daf743a514441d27b9 Mon Sep 17 00:00:00 2001 From: Jintao Huang Date: Fri, 15 Aug 2025 18:34:15 +0800 Subject: [PATCH] update docker --- docker/build_image.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/docker/build_image.py b/docker/build_image.py index 49ffd16d..d939e4e2 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -364,8 +364,7 @@ class SwiftImageBuilder(LLMImageBuilder): extra_content = extra_content.replace('{python_version}', self.args.python_version) extra_content += """ -RUN pip install --no-cache-dir -U deepspeed --no-deps && \ - pip install --no-cache-dir -U icecream soundfile pybind11 py-spy +RUN pip install --no-cache-dir -U icecream soundfile pybind11 py-spy """ version_args = ( f'{self.args.torch_version} {self.args.torchvision_version} {self.args.torchaudio_version} '