From f6e55485d99463b89cbd317396763369515616a3 Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Sun, 21 May 2023 07:01:34 +0000 Subject: [PATCH] Update gui.py --- gui.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/gui.py b/gui.py index 81d4d73..e80f4de 100644 --- a/gui.py +++ b/gui.py @@ -75,14 +75,14 @@ class RVC: self.version = cpt.get("version", "v1") if version == "v1": if if_f0 == 1: - net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half) + self.net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half) else: - net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"]) + self.net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"]) elif version == "v2": if if_f0 == 1: - net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half) + self.net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half) else: - net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"]) + self.net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"]) del self.net_g.enc_q print(self.net_g.load_state_dict(cpt["weight"], strict=False)) self.net_g.eval().to(device)