From f27562f1822bb5f42884ad8b1010d13e1b19f9df Mon Sep 17 00:00:00 2001 From: w-okada Date: Sat, 8 Jul 2023 06:12:49 +0900 Subject: [PATCH] Bugfix: new model add --- server/voice_changer/RVC/RVCModelSlotGenerator.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/server/voice_changer/RVC/RVCModelSlotGenerator.py b/server/voice_changer/RVC/RVCModelSlotGenerator.py index 6d1219f3..298c93e2 100644 --- a/server/voice_changer/RVC/RVCModelSlotGenerator.py +++ b/server/voice_changer/RVC/RVCModelSlotGenerator.py @@ -27,9 +27,9 @@ class RVCModelSlotGenerator(ModelSlotGenerator): # slotInfo.iconFile = "/assets/icons/noimage.png" if slotInfo.isONNX: - cls._setInfoByONNX(slotInfo) + slotInfo = cls._setInfoByONNX(slotInfo) else: - cls._setInfoByPytorch(slotInfo) + slotInfo = cls._setInfoByPytorch(slotInfo) return slotInfo @classmethod @@ -106,7 +106,10 @@ class RVCModelSlotGenerator(ModelSlotGenerator): for k, v in cpt["speaker_info"].items(): slot.speakers[int(k)] = str(v) + print("=========================> config::::::::::::1", cpt["config"]) + print("=========================> config::::::::::::2", cpt["config"][-1]) slot.samplingRate = cpt["config"][-1] + print("=========================> config::::::::::::3", slot.samplingRate) del cpt