From 59fa05d6e8b1ff8ed65b7a9e71b411eecd3927ff Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <stardust@fish.audio>
Date: Sat, 3 Aug 2024 16:02:52 +0800
Subject: [PATCH] Fix:(req) pinned torch version to 2.3.1, avoid inference
 speed abnormal slow down (#443)

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index bee88ff4..bca1ada9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,6 +32,7 @@ dependencies = [
     "loralib>=0.1.2",
     "natsort>=8.4.0",
     "pyrootutils>=1.0.4",
+    "torch==2.3.1",
     "vector_quantize_pytorch>=1.14.24",
     "resampy>=0.4.3",
     "einx[torch]==0.2.2",
@@ -41,7 +42,7 @@ dependencies = [
     "modelscope==1.16.1",
     "funasr==1.1.2",
     "opencc-python-reimplemented==0.1.7",
-    "torchaudio"
+    "torchaudio",
 ]
 
 [project.optional-dependencies]