From 204aed69ad5d99538b3caa155cbc54b5bb00b5f5 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Mon, 5 Feb 2024 21:04:47 +0900
Subject: [PATCH] Relax torch version to allow 2.1.x (useful for colab)

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 4cb8bad..fbf8489 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -24,6 +24,6 @@ requests
 safetensors
 scipy
 tensorboard
-torch==2.1.2 # For users without GPU
+torch>=2.1,<2.2 # For users without GPU or colab
 transformers
 umap-learn
\ No newline at end of file