PublicMirrors/voice-changer
1
0
Fork 0
mirror of https://github.com/w-okada/voice-changer.git synced 2025-03-14 20:03:59 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

WIP: DiffusionSVC

Browse Source
This commit is contained in:
w-okada 2023-07-15 23:24:37 +09:00
parent 2c4503ade8
commit 33e156ef9a
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/voice_changer/DiffusionSVC/DiffusionSVCModelSlotGenerator.py
View File

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