Spaces:
Runtime error
Runtime error
Update app_multi.py
Browse files- app_multi.py +22 -28
app_multi.py
CHANGED
@@ -103,35 +103,29 @@ for model_name in multi_cfg.get('models'):
|
|
103 |
else:
|
104 |
net_g = SynthesizerTrnMs768NSFsid_nono(*cpt['config'])
|
105 |
|
106 |
-
if 'enc_p.emb_phone.weight' in cpt['weight']:
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
upgraded_weight = torch.zeros(new_shape)
|
113 |
-
upgraded_weight[:, :old_shape[1]] = weight
|
114 |
-
cpt['weight']['enc_p.emb_phone.weight'] = upgraded_weight
|
115 |
-
|
116 |
-
del net_g.enc_q
|
117 |
-
|
118 |
-
# According to original code, this thing seems necessary.
|
119 |
-
print(net_g.load_state_dict(cpt['weight'], strict=False))
|
120 |
-
|
121 |
-
net_g.eval().to(config.device)
|
122 |
-
net_g = net_g.half() if util.is_half(config.device) else net_g.float()
|
123 |
-
|
124 |
-
vc = VC(tgt_sr, config)
|
125 |
-
|
126 |
-
loaded_models.append(dict(
|
127 |
-
name=model_name,
|
128 |
-
metadata=model_info,
|
129 |
-
vc=vc,
|
130 |
-
net_g=net_g,
|
131 |
-
if_f0=if_f0,
|
132 |
-
target_sr=tgt_sr
|
133 |
-
))
|
134 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
135 |
print(f'Models loaded: {len(loaded_models)}')
|
136 |
|
137 |
# Edge TTS speakers
|
|
|
103 |
else:
|
104 |
net_g = SynthesizerTrnMs768NSFsid_nono(*cpt['config'])
|
105 |
|
106 |
+
if 'enc_p.emb_phone.weight' not in cpt['weight']:
|
107 |
+
|
108 |
+
del net_g.enc_q
|
109 |
+
|
110 |
+
# According to original code, this thing seems necessary.
|
111 |
+
print(net_g.load_state_dict(cpt['weight'], strict=False))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
112 |
|
113 |
+
net_g.eval().to(config.device)
|
114 |
+
net_g = net_g.half() if util.is_half(config.device) else net_g.float()
|
115 |
+
|
116 |
+
vc = VC(tgt_sr, config)
|
117 |
+
|
118 |
+
loaded_models.append(dict(
|
119 |
+
name=model_name,
|
120 |
+
metadata=model_info,
|
121 |
+
vc=vc,
|
122 |
+
net_g=net_g,
|
123 |
+
if_f0=if_f0,
|
124 |
+
target_sr=tgt_sr
|
125 |
+
))
|
126 |
+
else:
|
127 |
+
print("Skip one module, because is too big")
|
128 |
+
|
129 |
print(f'Models loaded: {len(loaded_models)}')
|
130 |
|
131 |
# Edge TTS speakers
|