From 743304f3dad6bb4e7948720e5e3982195034ad91 Mon Sep 17 00:00:00 2001 From: tastelikefeet Date: Wed, 13 Aug 2025 15:22:02 +0800 Subject: [PATCH] fix dockerfile --- docker/Dockerfile.ubuntu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index f8d4222b..f72e4cc9 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 transformers -U; \ + pip install --no-cache-dir huggingface-hub transformers peft -U; \ fi; \ if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \ pip install liger_kernel nvitop pre-commit transformers huggingface-hub -U && \