From 3a92c046f90dd7316844e34ac98a9ebec2fb3d4e Mon Sep 17 00:00:00 2001 From: Kakaru <97896816+KakaruHayate@users.noreply.github.com> Date: Tue, 21 Oct 2025 17:11:54 +0800 Subject: [PATCH] revert "rfft fallback to cpu" --- GPT_SoVITS/eres2net/kaldi.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/GPT_SoVITS/eres2net/kaldi.py b/GPT_SoVITS/eres2net/kaldi.py index 4891f149..a80e5e6b 100644 --- a/GPT_SoVITS/eres2net/kaldi.py +++ b/GPT_SoVITS/eres2net/kaldi.py @@ -5,8 +5,6 @@ import torch import torchaudio from torch import Tensor -import musa_utils - __all__ = [ "get_mel_banks", "inverse_mel_scale", @@ -307,12 +305,7 @@ def spectrogram( ) # size (m, padded_window_size // 2 + 1, 2) - if musa_utils.is_available() : # 怎么还有算子不支持怎么还有算子不支持怎么还有算子不支持 - ori_device = strided_input.device - strided_input = strided_input.cpu() fft = torch.fft.rfft(strided_input) - if musa_utils.is_available() : - fft = fft.to(ori_device) # Convert the FFT into a power spectrum power_spectrum = torch.max(fft.abs().pow(2.0), epsilon).log() # size (m, padded_window_size // 2 + 1) @@ -625,12 +618,7 @@ def fbank( ) # size (m, padded_window_size // 2 + 1) - if musa_utils.is_available() : # 怎么还有算子不支持怎么还有算子不支持怎么还有算子不支持 - ori_device = strided_input.device - strided_input = strided_input.cpu() spectrum = torch.fft.rfft(strided_input).abs() - if musa_utils.is_available() : - spectrum = spectrum.to(ori_device) if use_power: spectrum = spectrum.pow(2.0)