diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 425f3f39..8269bfc2 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -79,7 +79,7 @@ RUN if [ "$INSTALL_MS_DEPS" = "True" ]; then \ pip install --no-cache-dir huggingface-hub transformers peft -U; \ fi; \ if [ "$INSTALL_MEGATRON_DEPS" = "True" ]; then \ - pip install "sglang[all]<0.5" "math_verify==0.5.2" "gradio<5.33" -U && \ + pip install "sglang[all]<0.5" "math_verify==0.5.2" "gradio<5.33" "qwen_vl_utils<0.0.12" -U && \ pip install liger_kernel wandb swanlab nvitop pre-commit "transformers<4.57" "trl<0.21" huggingface-hub -U && \ SITE_PACKAGES=$(python -c "import site; print(site.getsitepackages()[0])") && echo $SITE_PACKAGES && \ CUDNN_PATH=$SITE_PACKAGES/nvidia/cudnn CPLUS_INCLUDE_PATH=$SITE_PACKAGES/nvidia/cudnn/include \