Require torch>=2.1

2024-03-14 19:15:21 +09:00
parent 51e246658d
commit 1b72023e02
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -25,6 +25,6 @@ requests
 safetensors
 scipy
 tensorboard
-torch
+torch>=2.1
 transformers
 umap-learn