From 9b4cd6fe61b2fdcac487b2ef188ff702f77d3967 Mon Sep 17 00:00:00 2001
From: wangxingjun778 <wangxingjun778@163.com>
Date: Wed, 17 Sep 2025 20:42:06 +0800
Subject: [PATCH] update dockerfile for gpu image

---
 docker/Dockerfile.ubuntu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu
index acd8dc95..c0352614 100644
--- a/docker/Dockerfile.ubuntu
+++ b/docker/Dockerfile.ubuntu
@@ -76,7 +76,7 @@ RUN bash /tmp/install.sh {version_args} && \
 
 
 RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \
-    pip install --no-cache-dir huggingface-hub transformers peft ; \
+    pip install --no-cache-dir huggingface-hub transformers==4.55.4 peft ; \
 fi; \
 if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \
     pip install liger_kernel nvitop pre-commit transformers huggingface-hub && \