From 7230561cbd37d56bba07d20043f5c36895ee1710 Mon Sep 17 00:00:00 2001 From: suluyan Date: Thu, 5 Jun 2025 15:37:32 +0800 Subject: [PATCH] image deps upgrade: transformers, huggingface_hub, node.js --- docker/Dockerfile.ubuntu | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 2c444c31..2a7f96d8 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -76,7 +76,7 @@ RUN sh /tmp/install.sh {version_args} && \ RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \ - pip install --no-cache-dir huggingface-hub==0.25.* transformers==4.48.3 -U; \ + pip install --no-cache-dir huggingface-hub transformers -U; \ fi; \ if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \ pip install liger_kernel nvitop pre-commit transformers huggingface-hub -U && \ @@ -92,8 +92,8 @@ fi ENV NVM_DIR=/root/.nvm RUN curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.5/install.sh | bash && \ . $NVM_DIR/nvm.sh && \ - nvm install 18 && \ - nvm use 18 + nvm install 22 && \ + nvm use 22 ENV SETUPTOOLS_USE_DISTUTILS=stdlib ENV VLLM_USE_MODELSCOPE=True