Skip to content

Commit 497f266

Browse files
committed
Depend on the CPU version of torch
1 parent ea7bf6e commit 497f266

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

requirements.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,7 @@ flask
66
gunicorn
77
emoji
88
protobuf==3.20.*
9-
torch==2.5.1
9+
--extra-index-url https://download.pytorch.org/whl/cpu
10+
torch==2.5.1+cpu; sys_platform != "darwin" and platform_machine != "aarch64"
11+
torch==2.5.1; sys_platform == "darwin" or platform_machine == "aarch64"
1012
transformers==4.46.2

0 commit comments

Comments
 (0)