From aeb8fefc25ae9b75ac07d74bc399c6e26ab4ff67 Mon Sep 17 00:00:00 2001 From: Bingxu Chen Date: Mon, 11 May 2026 22:32:13 +0800 Subject: [PATCH] [AMD] Pin cache-dit==1.3.0 in rocm.Dockerfile + AMD CI install script (#24924) Co-authored-by: Cursor Agent --- 3rdparty/amd/wheel/sglang/pyproject.toml | 2 +- python/pyproject_other.toml | 2 +- scripts/ci/amd/amd_ci_install_dependency.sh | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/3rdparty/amd/wheel/sglang/pyproject.toml b/3rdparty/amd/wheel/sglang/pyproject.toml index d04c3f3bb..5ebdc6481 100644 --- a/3rdparty/amd/wheel/sglang/pyproject.toml +++ b/3rdparty/amd/wheel/sglang/pyproject.toml @@ -110,7 +110,7 @@ diffusion_hip = [ "st_attn==0.0.7", "vsa==0.0.4", "runai_model_streamer>=0.15.5", - "cache-dit==1.1.8", + "cache-dit==1.3.0", "addict", ] diff --git a/python/pyproject_other.toml b/python/pyproject_other.toml index d6f40474f..ad6f12e33 100755 --- a/python/pyproject_other.toml +++ b/python/pyproject_other.toml @@ -103,7 +103,7 @@ diffusion_hip = [ "st_attn==0.0.7", "vsa==0.0.4", "runai_model_streamer>=0.15.5", - "cache-dit==1.1.8", + "cache-dit==1.3.0", ] # For Intel Gaudi(device : hpu) follow the installation guide diff --git a/scripts/ci/amd/amd_ci_install_dependency.sh b/scripts/ci/amd/amd_ci_install_dependency.sh index d226d7ae5..9101fd3a5 100755 --- a/scripts/ci/amd/amd_ci_install_dependency.sh +++ b/scripts/ci/amd/amd_ci_install_dependency.sh @@ -174,7 +174,7 @@ EOF docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache pytest # Install cache-dit for qwen_image_t2i_cache_dit_enabled test (added in PR 16204) - docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache cache-dit || echo "cache-dit installation failed" + docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache --upgrade 'cache-dit==1.3.0' || echo "cache-dit installation failed" # Install accelerate for distributed training and inference support docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache accelerate || echo "accelerate installation failed"