Skip to content

Commit f6e5548

Browse files
authored
Update gui.py
1 parent 615c30c commit f6e5548

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

gui.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,14 +75,14 @@ def __init__(
7575
self.version = cpt.get("version", "v1")
7676
if version == "v1":
7777
if if_f0 == 1:
78-
net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half)
78+
self.net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=config.is_half)
7979
else:
80-
net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
80+
self.net_g = SynthesizerTrnMs256NSFsid_nono(*cpt["config"])
8181
elif version == "v2":
8282
if if_f0 == 1:
83-
net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half)
83+
self.net_g = SynthesizerTrnMs768NSFsid(*cpt["config"], is_half=config.is_half)
8484
else:
85-
net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
85+
self.net_g = SynthesizerTrnMs768NSFsid_nono(*cpt["config"])
8686
del self.net_g.enc_q
8787
print(self.net_g.load_state_dict(cpt["weight"], strict=False))
8888
self.net_g.eval().to(device)

0 commit comments

Comments
 (0)