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

WIP: refactoring

Browse Source
This commit is contained in:
wataru 2023-04-28 10:37:55 +09:00
parent 443fe1e720
commit 37bb25682e
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
server/voice_changer/RVC/RVC.py
View File

@ -406,14 +406,10 @@ class RVC:
repeat *= self.settings.rvcQuality # 0 or 3 repeat *= self.settings.rvcQuality # 0 or 3
vc = VC(self.settings.modelSamplingRate, dev, self.is_half, repeat) vc = VC(self.settings.modelSamplingRate, dev, self.is_half, repeat)
sid = 0 sid = 0
times = [0, 0, 0]
f0_up_key = self.settings.tran f0_up_key = self.settings.tran
f0_method = self.settings.f0Detector f0_method = self.settings.f0Detector
file_index = self.index_file if self.index_file is not None else ""
file_big_npy = self.feature_file if self.feature_file is not None else ""
index_rate = self.settings.indexRatio index_rate = self.settings.indexRatio
if_f0 = 1 if self.settings.modelSlots[self.currentSlot].f0 else 0 if_f0 = 1 if self.settings.modelSlots[self.currentSlot].f0 else 0
f0_file = None
embChannels = self.settings.modelSlots[self.currentSlot].embChannels embChannels = self.settings.modelSlots[self.currentSlot].embChannels
audio_out = vc.pipeline( audio_out = vc.pipeline(
@ -421,14 +417,12 @@ class RVC:
self.onnx_session, self.onnx_session,
sid, sid,
audio, audio,
times,
f0_up_key, f0_up_key,
f0_method, f0_method,
file_index, self.index,
file_big_npy, self.feature,
index_rate, index_rate,
if_f0, if_f0,
f0_file=f0_file,
silence_front=self.settings.extraConvertSize silence_front=self.settings.extraConvertSize
/ self.settings.modelSamplingRate, / self.settings.modelSamplingRate,
embChannels=embChannels, embChannels=embChannels,
@ -453,14 +447,13 @@ class RVC:
else: else:
dev = torch.device("cuda", index=self.settings.gpu) dev = torch.device("cuda", index=self.settings.gpu)
# print("device:", dev)
self.hubert_model = self.hubert_model.to(dev) self.hubert_model = self.hubert_model.to(dev)
self.net_g = self.net_g.to(dev) self.net_g = self.net_g.to(dev)
audio = data[0] audio = data[0]
convertSize = data[1] convertSize = data[1]
vol = data[2] vol = data[2]
audio = resampy.resample(audio, self.settings.modelSamplingRate, 16000) audio = resampy.resample(audio, self.settings.modelSamplingRate, 16000)
if vol < self.settings.silentThreshold: if vol < self.settings.silentThreshold: