Skip to content

Commit 9470fba

Browse files
authored
ci: Fix vllm version (#973)
* Fix vllm version Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com> * Transformers 4.53.1 Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com> --------- Signed-off-by: Dong Hyuk Chang <donghyukc@nvidia.com>
1 parent 83d43fd commit 9470fba

File tree

2 files changed

+196
-204
lines changed

2 files changed

+196
-204
lines changed

ray-curator/pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ video = [
7878
video_cuda = [
7979
"cvcuda_cu12",
8080
"pycuda",
81-
"vllm; (platform_machine == 'x86_64' and platform_system != 'Darwin')",
81+
"vllm==0.9.2; (platform_machine == 'x86_64' and platform_system != 'Darwin')",
8282
"flash-attn<=2.8.3; (platform_machine == 'x86_64' and platform_system != 'Darwin')",
8383
"torch<=2.8.0",
8484
"torchaudio",

0 commit comments

Comments
 (0)