minor fix

SWivid · Oct 22, 2024 · 198d44d · 198d44d
1 parent 752f6f5
commit 198d44d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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 = []