From db369fda000234d117abfcea9a50fdc418987945 Mon Sep 17 00:00:00 2001
From: "yuze.zyz" <yuze.zyz@alibaba-inc.com>
Date: Mon, 28 Oct 2024 18:25:26 +0800
Subject: [PATCH] fix tests

---
 docker/Dockerfile.ubuntu | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu
index 056c0026..ee5ff2cd 100644
--- a/docker/Dockerfile.ubuntu
+++ b/docker/Dockerfile.ubuntu
@@ -46,6 +46,8 @@ 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
+
 RUN pip config set global.index-url https://mirrors.aliyun.com/pypi/simple && \
     pip config set install.trusted-host mirrors.aliyun.com && \
     cp /tmp/resources/ubuntu2204.aliyun /etc/apt/sources.list
@@ -54,3 +56,4 @@ ENV SETUPTOOLS_USE_DISTUTILS=stdlib
 ENV VLLM_USE_MODELSCOPE=True
 ENV LMDEPLOY_USE_MODELSCOPE=True
 ENV MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope
+CMD ["bash"]