File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -75,14 +75,14 @@ def __init__(
75
75
self .version = cpt .get ("version" , "v1" )
76
76
if version == "v1" :
77
77
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 )
79
79
else :
80
- net_g = SynthesizerTrnMs256NSFsid_nono (* cpt ["config" ])
80
+ self . net_g = SynthesizerTrnMs256NSFsid_nono (* cpt ["config" ])
81
81
elif version == "v2" :
82
82
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 )
84
84
else :
85
- net_g = SynthesizerTrnMs768NSFsid_nono (* cpt ["config" ])
85
+ self . net_g = SynthesizerTrnMs768NSFsid_nono (* cpt ["config" ])
86
86
del self .net_g .enc_q
87
87
print (self .net_g .load_state_dict (cpt ["weight" ], strict = False ))
88
88
self .net_g .eval ().to (device )
You can’t perform that action at this time.
0 commit comments