Sync from GitHub repo

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

model/utils_infer.py

@@ -24,7 +24,6 @@ from model.utils import (
 
 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 = []