From 198d44db650fa4c4a60c562cecd3fe0b090fd399 Mon Sep 17 00:00:00 2001
From: SWivid <swivid@qq.com>
Date: Tue, 22 Oct 2024 17:54:54 +0800
Subject: [PATCH] minor fix

---
 model/utils_infer.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/model/utils_infer.py b/model/utils_infer.py
index 9c1a4dbc6..03552206a 100644
--- a/model/utils_infer.py
+++ b/model/utils_infer.py
@@ -24,7 +24,6 @@
 
 def get_device():
     device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
-    # print(f"Using {device} device")
     return device
 
 
@@ -273,7 +272,7 @@ def infer_batch_process(
     if sr != target_sample_rate:
         resampler = torchaudio.transforms.Resample(sr, target_sample_rate)
         audio = resampler(audio)
-    audio = audio.to()
+    audio = audio.to(device)
 
     generated_waves = []
     spectrograms = []