From 51e246658d25631d2498cd6cbe79b6906595ee21 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Thu, 14 Mar 2024 18:30:21 +0900 Subject: [PATCH] Change torch==2.1 to torch >=2.1 --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index bf47ec0..680f28b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -25,6 +25,6 @@ requests safetensors scipy tensorboard -torch>=2.1,<2.2 +torch transformers umap-learn