From 189d022967d05984b103e33a421e88195174c422 Mon Sep 17 00:00:00 2001 From: wangxingjun778 Date: Wed, 17 Sep 2025 15:34:51 +0800 Subject: [PATCH] fix transformers version for gpu image --- docker/Dockerfile.ubuntu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 5e1a1b6b..14715b89 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -68,7 +68,7 @@ RUN bash /tmp/install.sh {version_args} && \ cd modelscope && pip install . -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html && \ cd / && rm -fr /tmp/modelscope && pip cache purge; \ pip install --no-cache-dir torch=={torch_version} torchvision=={torchvision_version} torchaudio=={torchaudio_version} {index_url} && \ - pip install --no-cache-dir transformers diffusers timm>=0.9.0 && pip cache purge; \ + pip install --no-cache-dir transformers==4.55.4 diffusers timm>=0.9.0 && pip cache purge; \ pip install --no-cache-dir omegaconf==2.3.0 && pip cache purge; \ pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \ pip config set install.trusted-host mirrors.aliyun.com && \