new model add
This commit is contained in:
w-okada 2023-07-08 06:12:49 +09:00
parent cb9043ffd4
commit f27562f182

View File

@ -27,9 +27,9 @@ class RVCModelSlotGenerator(ModelSlotGenerator):
# slotInfo.iconFile = "/assets/icons/noimage.png" # slotInfo.iconFile = "/assets/icons/noimage.png"
if slotInfo.isONNX: if slotInfo.isONNX:
cls._setInfoByONNX(slotInfo) slotInfo = cls._setInfoByONNX(slotInfo)
else: else:
cls._setInfoByPytorch(slotInfo) slotInfo = cls._setInfoByPytorch(slotInfo)
return slotInfo return slotInfo
@classmethod @classmethod
@ -106,7 +106,10 @@ class RVCModelSlotGenerator(ModelSlotGenerator):
for k, v in cpt["speaker_info"].items(): for k, v in cpt["speaker_info"].items():
slot.speakers[int(k)] = str(v) slot.speakers[int(k)] = str(v)
print("=========================> config::::::::::::1", cpt["config"])
print("=========================> config::::::::::::2", cpt["config"][-1])
slot.samplingRate = cpt["config"][-1] slot.samplingRate = cpt["config"][-1]
print("=========================> config::::::::::::3", slot.samplingRate)
del cpt del cpt