Skip to content

Commit a906079

Browse files
author
chibu
committed
pin lib versions
1 parent 07fc331 commit a906079

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/automation/tasks/guidellm.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,9 @@
99
class GuideLLMTask(BaseTask):
1010

1111
guidellm_packages = [
12-
"vllm",
12+
"transformers==4.53.0",
13+
"huggingface-hub==0.34.3",
14+
"vllm==0.9.1",
1315
GUIDELLM_PACKAGE,
1416
"hf_xet",
1517
]

0 commit comments

Comments
 (0)