Depend on the CPU version of torch

GateNLP · Nov 18, 2024 · 497f266 · 497f266
1 parent ea7bf6e
commit 497f266
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -6,5 +6,7 @@ flask
 gunicorn
 emoji
 protobuf==3.20.*
-torch==2.5.1
+--extra-index-url https://download.pytorch.org/whl/cpu
+torch==2.5.1+cpu; sys_platform != "darwin" and platform_machine != "aarch64"
+torch==2.5.1; sys_platform == "darwin" or platform_machine == "aarch64"
 transformers==4.46.2