[AMD] Pin cache-dit==1.3.0 in rocm.Dockerfile + AMD CI install script (#24924)
Co-authored-by: Cursor Agent <cursoragent@cursor.com>
This commit is contained in:
co-authored by
Cursor Agent
parent
d9cb38012e
commit
aeb8fefc25
+1
-1
@@ -110,7 +110,7 @@ diffusion_hip = [
|
|||||||
"st_attn==0.0.7",
|
"st_attn==0.0.7",
|
||||||
"vsa==0.0.4",
|
"vsa==0.0.4",
|
||||||
"runai_model_streamer>=0.15.5",
|
"runai_model_streamer>=0.15.5",
|
||||||
"cache-dit==1.1.8",
|
"cache-dit==1.3.0",
|
||||||
"addict",
|
"addict",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|||||||
@@ -103,7 +103,7 @@ diffusion_hip = [
|
|||||||
"st_attn==0.0.7",
|
"st_attn==0.0.7",
|
||||||
"vsa==0.0.4",
|
"vsa==0.0.4",
|
||||||
"runai_model_streamer>=0.15.5",
|
"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
|
# For Intel Gaudi(device : hpu) follow the installation guide
|
||||||
|
|||||||
@@ -174,7 +174,7 @@ EOF
|
|||||||
docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache pytest
|
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)
|
# 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
|
# 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"
|
docker exec ci_sglang pip install --cache-dir=/sgl-data/pip-cache accelerate || echo "accelerate installation failed"
|
||||||
|
|||||||
Reference in New Issue
Block a user