set transformers==4.44.2 in dockerfile

This commit is contained in:
xingjun.wang
2024-10-30 19:53:23 +08:00
parent a9cec462dc
commit eb526aaa52

View File

@@ -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 && \