From 8b2bd819324d7def592fee4c7c19805038525aca Mon Sep 17 00:00:00 2001
From: lich99 <lichenghai99@gmail.com>
Date: Fri, 31 May 2024 12:51:33 +0800
Subject: [PATCH] update torch version

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

diff --git a/requirements.txt b/requirements.txt
index 87868ec..ac915fb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 omegaconf~=2.3.0
-torch~=2.0
+torch~=2.1.0
 tqdm
 einops
 vector_quantize_pytorch