diff --git a/docker/runtime/ai/base/requirements.txt b/docker/runtime/ai/base/requirements.txt index f1cd5c86b..da24ce3a8 100644 --- 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 index eacbfb2a7..90d202057 100644 --- 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 index f97913fbf..2f2c04af0 100644 --- 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