fix

underworld02 · Sep 13, 2022 · 3aabb9e · 3aabb9e
1 parent 2adc2e8
commit 3aabb9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MoeGoe.py b/MoeGoe.py
@@ -180,7 +180,7 @@ def get_label(text,label):
             with no_grad():
                 x_tst = stn_tst.unsqueeze(0)
                 x_tst_lengths = LongTensor([stn_tst.size(0)])
-                sid = LongTensor([0])
+                sid = LongTensor([target_id])
                 audio = net_g_ms.infer(x_tst, x_tst_lengths, sid=sid, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=length_scale)[0][0, 0].data.float().numpy()
             write(out_path, hps_ms.data.sampling_rate, audio)