From 1d9b91fd2a92d202118394bc2f7430a4720c358f Mon Sep 17 00:00:00 2001 From: litagin02 <139731664+litagin02@users.noreply.github.com> Date: Thu, 14 Mar 2024 19:02:55 +0900 Subject: [PATCH] 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. --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 6bf329f..b0580e3 100644 --- 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