AI: upgrade the transformers to 4.30.2 for CVE fix (#1620)

oap-project · Jul 4, 2023 · 5b218d4 · 5b218d4
1 parent cc3857f
commit 5b218d4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docker/runtime/ai/base/requirements.txt b/docker/runtime/ai/base/requirements.txt
@@ -19,7 +19,7 @@ xgboost==1.7.5
 lightgbm==3.3.5
 
 # DL Frameworks
-transformers==4.29.1
+transformers==4.30.2
 
 # datasets and models
 datasets~=2.9.0

diff --git a/python/cloudtik/runtime/ai/scripts/install.sh b/python/cloudtik/runtime/ai/scripts/install.sh
@@ -57,7 +57,7 @@ function install_ml() {
         hyperopt==0.2.7 \
         scikit-learn==1.0.2 \
         xgboost==1.7.5 \
-        transformers==4.29.1 \
+        transformers==4.30.2 \
         pandas==2.0.1 \
         category-encoders==2.6.0 \
         h5py==3.8.0 \

diff --git a/python/cloudtik/runtime/common/scripts/cloud-storage-fuse.sh b/python/cloudtik/runtime/common/scripts/cloud-storage-fuse.sh
@@ -258,7 +258,7 @@ function mount_local_hdfs_fs() {
     else
         FS_MOUNT_PATH=${CLUSTER_FS_MOUNT_PATH}
     fi
-    # Mount local hdfs fuse here
+    # Mount local hdfs here
     mkdir -p ${FS_MOUNT_PATH}
 
     if [ "${HDFS_MOUNT_METHOD}" == "nfs" ]; then