Update torch req to torch>=2.1 from torch==2.1.x

because this req is only needed in colab or linux, and it's no problem for torch 2.2.x for linux.
2024-03-14 19:02:55 +09:00
parent 9292270873
commit 1d9b91fd2a
1 changed files with 1 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,6 @@ requests
 safetensors
 scipy
 tensorboard
-torch>=2.1,<2.2
+torch>=2.1
 transformers
 umap-learn