WIP: DiffusionSVC

This commit is contained in:
w-okada 2023-07-15 23:24:37 +09:00
parent 2c4503ade8
commit 33e156ef9a

View File

@ -33,6 +33,8 @@ class DiffusionSVCModelSlotGenerator(ModelSlotGenerator):
def _setInfoByPytorch(cls, slot: DiffusionSVCModelSlot):
diff_model, diff_args, naive_model, naive_args, vocoder = load_model_vocoder_from_combo(slot.modelFile, device="cpu")
slot.kStepMax = diff_args.model.k_step_max
slot.n_layers = diff_args.model.n_layers
return slot
@classmethod