From eb526aaa52e2492871953e2de3d3e69a21a200ee Mon Sep 17 00:00:00 2001 From: "xingjun.wang" Date: Wed, 30 Oct 2024 19:53:23 +0800 Subject: [PATCH] set transformers==4.44.2 in dockerfile --- docker/Dockerfile.ubuntu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 1e43c3fd..efd62c17 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -46,7 +46,7 @@ RUN cd /tmp && GIT_LFS_SKIP_SMUDGE=1 git clone -b {swift_branch} --single-branc RUN pip install --no-cache-dir torch=={torch_version} torchvision=={torchvision_version} torchaudio=={torchaudio_version} {index_url} -RUN pip install --no-cache-dir transformers -U huggingface-hub==0.25.0 +RUN pip install --no-cache-dir transformers==4.44.2 huggingface-hub==0.25.0 RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \ pip config set install.trusted-host mirrors.aliyun.com && \