From cb9b78af3080b01de0ff4cd9f331fed3d5c51dbe Mon Sep 17 00:00:00 2001
From: AIFSH <1509359472@qq.com>
Date: Sun, 25 Aug 2024 08:04:47 +0800
Subject: [PATCH] fix bug

---
 __init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 4cd10da..297818d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -147,8 +147,7 @@ class VocalSeparationNode:
                     estimates = estimates * std + mean
         
         estimates_mono = estimates.mean(1)
-        estimates_t = torch.Tensor(estimates_mono).unsqueeze(0).unsqueeze(0)
-        # print(estimates_t.shape)
+        estimates_t = torch.tensor(estimates_mono,dtype=torch.float32).unsqueeze(0).unsqueeze(0)
         vocals = {
             "waveform":estimates_t,
             "sample_rate": target_sr
@@ -160,6 +159,7 @@ class VocalSeparationNode:
             "sample_rate": target_sr
         }
         self.model.to("cpu")
+        torch.backends.cudnn.benchmark = False
         return (vocals,instrumental,)
 
 NODE_CLASS_MAPPINGS = {