Merge pull request #10 from rodion-m/patch-1

Fix synthesizing with cpu (#8)
IS2AI · Feb 23, 2023 · a3869c4 · a3869c4
2 parents d28354e + 532b1ac
commit a3869c4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tts1/synthesize.py b/tts1/synthesize.py
@@ -43,8 +43,9 @@ def get_args():
 args = get_args()
 sample_text = args.text
 with torch.no_grad():
-    _, c_mel, *_ = text2speech(sample_text.lower())
-    wav = vocoder.inference(c_mel)
+    output_dict = text2speech(sample_text.lower())
+    feat_gen = output_dict['feat_gen']
+    wav = vocoder.inference(feat_gen)
 
 ## here all of your synthesized audios will be saved
 folder_to_save, wav_name = "synthesized_wavs", "example.wav"