fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-08 03:55:47 +08:00
Code Activity

fix(rtrvc): parameter issue

Browse Source
This commit is contained in:
源文雨
2024-06-16 17:25:01 +09:00
parent 0d5cd347bc
commit 3a79d81907
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
infer/lib/rtrvc.py
View File

@@ -213,8 +213,7 @@ class RVC:
pitch, pitchf = self._get_f0( pitch, pitchf = self._get_f0(
input_wav[-f0_extractor_frame:], input_wav[-f0_extractor_frame:],
self.f0_up_key - self.formant_shift, self.f0_up_key - self.formant_shift,
3, method=f0method,
f0method,
) )
shift = block_frame_16k // self.window shift = block_frame_16k // self.window
self.cache_pitch[:-shift] = self.cache_pitch[shift:].clone() self.cache_pitch[:-shift] = self.cache_pitch[shift:].clone()
@@ -297,7 +296,7 @@ class RVC:
def _get_f0_pm(self, x, f0_up_key, filter_radius): def _get_f0_pm(self, x, f0_up_key, filter_radius):
if not hasattr(self, "pm"): if not hasattr(self, "pm"):
self.pm = PM(hop_length=160, sampling_rate=16000) self.pm = PM(hop_length=160, sampling_rate=16000)
f0 = self.pm.compute_f0(x) f0 = self.pm.compute_f0(x.cpu().numpy())
return self._get_f0_post(f0, f0_up_key) return self._get_f0_post(f0, f0_up_key)
def _get_f0_harvest(self, x, f0_up_key, filter_radius=3): def _get_f0_harvest(self, x, f0_up_key, filter_radius=3):
@@ -309,7 +308,7 @@ class RVC:
self.sr, self.sr,
) )
if filter_radius is None: filter_radius=3 if filter_radius is None: filter_radius=3
f0 = self.harvest.compute_f0(x, filter_radius=filter_radius) f0 = self.harvest.compute_f0(x.cpu().numpy(), filter_radius=filter_radius)
return self._get_f0_post(f0, f0_up_key) return self._get_f0_post(f0, f0_up_key)
def _get_f0_dio(self, x, f0_up_key, filter_radius): def _get_f0_dio(self, x, f0_up_key, filter_radius):
@@ -320,7 +319,7 @@ class RVC:
self.f0_max, self.f0_max,
self.sr, self.sr,
) )
f0 = self.dio.compute_f0(x) f0 = self.dio.compute_f0(x.cpu().numpy())
return self._get_f0_post(f0, f0_up_key) return self._get_f0_post(f0, f0_up_key)
def _get_f0_crepe(self, x, f0_up_key, filter_radius): def _get_f0_crepe(self, x, f0_up_key, filter_radius):

2
rvc/layers/synthesizers.py
View File

@@ -188,7 +188,7 @@ class SynthesizerTrnMsNSFsid(nn.Module):
flow_head = head - 24 flow_head = head - 24
if flow_head < 0: flow_head = 0 if flow_head < 0: flow_head = 0
dec_head = head - flow_head dec_head = head - flow_head
m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths, head) m_p, logs_p, x_mask = self.enc_p(phone, pitch, phone_lengths, flow_head)
z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask z_p = (m_p + torch.exp(logs_p) * torch.randn_like(m_p) * 0.66666) * x_mask
z = self.flow(z_p, x_mask, g=g, reverse=True) z = self.flow(z_p, x_mask, g=g, reverse=True)
z = z[:, :, dec_head : dec_head + length] z = z[:, :, dec_head : dec_head + length]