File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 )
You can’t perform that action at this time.
0 commit comments