From 33e156ef9a5617b8f810ef88cab4eb2b5c0e8a6e Mon Sep 17 00:00:00 2001 From: w-okada Date: Sat, 15 Jul 2023 23:24:37 +0900 Subject: [PATCH] WIP: DiffusionSVC --- .../DiffusionSVC/DiffusionSVCModelSlotGenerator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/server/voice_changer/DiffusionSVC/DiffusionSVCModelSlotGenerator.py b/server/voice_changer/DiffusionSVC/DiffusionSVCModelSlotGenerator.py index 3940846d..86f1ca22 100644 --- a/server/voice_changer/DiffusionSVC/DiffusionSVCModelSlotGenerator.py +++ b/server/voice_changer/DiffusionSVC/DiffusionSVCModelSlotGenerator.py @@ -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