2023-11-12 23:10:58 +09:00
|
|
|
import os
|
|
|
|
|
|
2024-02-28 23:08:49 +09:00
|
|
|
from data.ModelSlot import LLVCModelSlot
|
2023-11-12 23:10:58 +09:00
|
|
|
from voice_changer.utils.LoadModelParams import LoadModelParams
|
|
|
|
|
from voice_changer.utils.ModelSlotGenerator import ModelSlotGenerator
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class LLVCModelSlotGenerator(ModelSlotGenerator):
|
|
|
|
|
@classmethod
|
|
|
|
|
def loadModel(cls, props: LoadModelParams):
|
|
|
|
|
slotInfo: LLVCModelSlot = LLVCModelSlot()
|
|
|
|
|
for file in props.files:
|
|
|
|
|
if file.kind == "llvcModel":
|
|
|
|
|
slotInfo.modelFile = file.name
|
|
|
|
|
if file.kind == "llvcConfig":
|
|
|
|
|
slotInfo.configFile = file.name
|
|
|
|
|
slotInfo.name = os.path.splitext(os.path.basename(slotInfo.modelFile))[0]
|
|
|
|
|
slotInfo.slotIndex = props.slot
|
|
|
|
|
return slotInfo
|
