gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-03 00:30:51 -08:00
Code Issues Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
SWivid
2024-10-22 17:54:54 +08:00
parent 752f6f5ea8
commit 198d44db65
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
model/utils_infer.py
View File

@@ -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 = []