1berry/wokada-voice-changer
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

WIP: multi model

Browse Source
This commit is contained in:
wataru 2023-04-22 07:57:51 +09:00
parent f8b589cc1d
commit 1fd3f04030
1 changed files with 15 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -82,6 +82,7 @@ class RVCSettings():
class RVC:
def __init__(self, params):
self.initialLoad = True
self.settings = RVCSettings()
self.inferenceing: bool = False
@ -133,8 +134,12 @@ class RVC:
print("EXCEPTION during loading hubert/contentvec model", e)
# self.switchModel(self.slot)
self.prepareModel(self.tmp_slot)
self.slot = self.tmp_slot
if self.initialLoad:
self.prepareModel(self.tmp_slot)
self.slot = self.tmp_slot
self.currentSlot = self.slot
self.switchModel()
self.initialLoad = False
return self.get_info()
@ -346,31 +351,16 @@ class RVC:
return result
def inference(self, data):
if self.inferenceing == True:
raise NoModeLoadedException("---------------------------------- tmp")
if self.currentSlot != self.slot:
self.currentSlot = self.slot
self.switchModel()
self.inferenceing = True
if self.settings.framework == "ONNX":
audio = self._onnx_inference(data)
else:
audio = self._pyTorch_inference(data)
try:
if self.currentSlot != self.slot:
import time
# time.sleep(1)
self.currentSlot = self.slot
# self.prepareModel(self.currentSlot)
self.switchModel()
# time.sleep(1)
if self.settings.framework == "ONNX":
audio = self._onnx_inference(data)
else:
audio = self._pyTorch_inference(data)
self.inferenceing = False
return audio
except Exception as e:
self.inferenceing = False
raise e
return audio
def __del__(self):
del self.net_g