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