fix(dependencies): pin pytorch version (#3872)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
2025-05-20 10:35:01 +00:00 · 2024-10-18 09:11:59 +02:00 · 2024-10-18 09:11:59 +02:00 · 3e77a17b26
commit 3e77a17b26
parent a26fb548b1
46 changed files with 58 additions and 58 deletions
--- a/backend/python/exllama2/requirements-cpu.txt
+++ b/backend/python/exllama2/requirements-cpu.txt
@ -1,3 +1,3 @@
 transformers
 accelerate
-torch
+torch==2.4.1
--- a/backend/python/exllama2/requirements-cublas11.txt
+++ b/backend/python/exllama2/requirements-cublas11.txt
@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cu118
-torch
+torch==2.4.1+cu118
 transformers
 accelerate
--- a/backend/python/exllama2/requirements-cublas12.txt
+++ b/backend/python/exllama2/requirements-cublas12.txt
@ -1,3 +1,3 @@
-torch
+torch==2.4.1
 transformers
 accelerate