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

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto authored and GitHub committed 2024-10-18 09:11:59 +02:00
1 parent a26fb548b1
commit 3e77a17b26
46 files changed
+58 -58

No files matched your search

+1 -1
View File
@@ -1,3 +1,3 @@
accelerate
torch
torch==2.4.1
transformers
@@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/cu118
accelerate
torch
torch==2.4.1+cu118
transformers
bitsandbytes
@@ -1,4 +1,4 @@
accelerate
torch
torch==2.4.1
transformers
bitsandbytes
+1 -1
View File
@@ -1,5 +1,5 @@
--extra-index-url https://download.pytorch.org/whl/rocm6.0
accelerate
torch
torch==2.4.1+rocm6.0
transformers
bitsandbytes