From c176105761f4c2fb965ae258da4f164ab5cd64c1 Mon Sep 17 00:00:00 2001
From: desilo-hanyul <95674196+hanyul-ryu@users.noreply.github.com>
Date: Tue, 6 Aug 2024 13:32:26 +0900
Subject: [PATCH] Update pyproject.toml

fix torch version
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 7b8e29c..f4ad917 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,7 +54,7 @@ mpmath = "^1.3.0"
 scipy = "^1.10.1"
 matplotlib = "^3.7.1"
 joblib = "^1.2.0"
-torch = "^2.0.0"
+torch = "==2.1.1"
 tqdm = "^4.66.1"
 
 [tool.poetry.group.dev.dependencies]