[CI] Fix CI timeouts by upgrading runai_model_streamer (related to #16937) (#17636)

This commit is contained in:
Prozac614
2026-01-29 09:09:45 +08:00
committed by GitHub
parent d4180815a4
commit 3fcda00e8c
3 changed files with 7 additions and 3 deletions

View File

@@ -95,7 +95,7 @@ diffusion = [
"remote-pdb",
"st_attn==0.0.7 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
"vsa==0.0.4 ; platform_machine != 'aarch64' and platform_machine != 'arm64'",
"runai_model_streamer",
"runai_model_streamer>=0.15.5",
"cache-dit==1.2.0"
]

View File

@@ -96,7 +96,7 @@ diffusion_hip = [
"torchcodec==0.5.0",
"st_attn==0.0.7",
"vsa==0.0.4",
"runai_model_streamer",
"runai_model_streamer>=0.15.5",
]
# For Intel Gaudi(device : hpu) follow the installation guide
@@ -124,7 +124,7 @@ diffusion_musa = [
"remote-pdb",
"st_attn==0.0.7",
"vsa==0.0.4",
"runai_model_streamer",
"runai_model_streamer>=0.15.5",
"cache-dit==1.1.8"
]