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.
litagin02 · Mar 14, 2024 · 1d9b91f · 1d9b91f
1 parent 9292270
commit 1d9b91f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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