Fix torch 2.2.0 bug to use 2.1.2

2024-02-02 22:09:57 +09:00
parent ceee581575
commit c22c9cd142
2 changed files with 5 additions and 6 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,6 @@ requests
 safetensors
 scipy
 tensorboard
-torch # For users without GPU
+torch==2.1.2 # For users without GPU
 transformers
 umap-learn