PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-02-02 16:23:58 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

RMVPE:

Browse Source 
different device bug, not finding root caused yet.
This commit is contained in:
w-okada 2023-07-27 04:12:58 +09:00
parent f96b4c2414
commit 9de74281f6
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/voice_changer/DiffusionSVC/pitchExtractor/rmvpe/rmvpe.py
View File

@ -4,6 +4,10 @@ import torch
import numpy as np
from librosa.filters import mel
from mods.log_control import VoiceChangaerLogger
logger = VoiceChangaerLogger.get_instance().getLogger()
class BiGRU(nn.Module):
def __init__(self, input_features, hidden_features, num_layers):
@ -318,6 +322,9 @@ class MelSpectrogram(torch.nn.Module):
if resize < size:
magnitude = F.pad(magnitude, (0, 0, 0, size - resize))
magnitude = magnitude[:, :size, :] * self.win_length / win_length_new
if self.mel_basis.device != magnitude.device:
logger.warn(f"[RMVPE] Device is not same. mel_basis:{self.mel_basis.device}, magnitude:{magnitude.device}")
self.mel_basis.to(magnitude.device)
mel_output = torch.matmul(self.mel_basis, magnitude)
if self.is_half is True:
mel_output = mel_output.half()