Skip to content

Commit 6e3a57f

Browse files
committed
Lock more deps
Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com>
1 parent 7d5266f commit 6e3a57f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,8 +66,8 @@
6666
],
6767
extras_require={
6868
"llm": [
69-
"torch>=2.0.0",
70-
"transformers",
69+
"torch>=2.6.0",
70+
"transformers<=4.51.3",
7171
"accelerate",
7272
"py-cpuinfo",
7373
"sentencepiece",
@@ -77,7 +77,7 @@
7777
"human-eval-windows==1.0.4",
7878
"fastapi",
7979
"uvicorn[standard]",
80-
"openai>=1.66.0",
80+
"openai>=1.81.0",
8181
"lm-eval[api]",
8282
],
8383
"llm-oga-cpu": [

0 commit comments

Comments
 (0)