From e2ee8430988485c1c82f94192de24a596b3a4eb0 Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 12 Apr 2022 16:40:42 -0600 Subject: [PATCH] Updates --- api.py | 2 +- eval_multiple.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api.py b/api.py index b1f3740..f5b2cd6 100644 --- a/api.py +++ b/api.py @@ -138,7 +138,7 @@ class TextToSpeech: heads=16, number_text_tokens=256, start_text_token=255, checkpointing=False, train_solo_embeddings=False, average_conditioning_embeddings=True).cpu().eval() - self.autoregressive.load_state_dict(torch.load('.models/autoregressive.pth')) + self.autoregressive.load_state_dict(torch.load('.models/autoregressive_diverse.pth')) self.clip = VoiceCLIP(dim_text=512, dim_speech=512, dim_latent=512, num_text_tokens=256, text_enc_depth=12, text_seq_len=350, text_heads=8, diff --git a/eval_multiple.py b/eval_multiple.py index b772f8a..c55cdc1 100644 --- a/eval_multiple.py +++ b/eval_multiple.py @@ -7,7 +7,7 @@ from utils.audio import load_audio if __name__ == '__main__': fname = 'Y:\\libritts\\test-clean\\transcribed-brief-w2v.tsv' - outpath = 'D:\\tmp\\tortoise-tts-eval\\eval_new_autoregressive' + outpath = 'D:\\tmp\\tortoise-tts-eval\\diverse_auto_256_samp_100_di_4' outpath_real = 'D:\\tmp\\tortoise-tts-eval\\real' os.makedirs(outpath, exist_ok=True) @@ -24,9 +24,9 @@ if __name__ == '__main__': path = os.path.join(os.path.dirname(fname), line[1]) cond_audio = load_audio(path, 22050) torchaudio.save(os.path.join(outpath_real, os.path.basename(line[1])), cond_audio, 22050) - sample = tts.tts(transcript, [cond_audio, cond_audio], num_autoregressive_samples=512, k=1, + sample = tts.tts(transcript, [cond_audio, cond_audio], num_autoregressive_samples=256, k=1, repetition_penalty=2.0, length_penalty=2, temperature=.5, top_p=.5, - diffusion_temperature=.7, cond_free_k=2, diffusion_iterations=200) + diffusion_temperature=.7, cond_free_k=2, diffusion_iterations=100) down = torchaudio.functional.resample(sample, 24000, 22050) fout_path = os.path.join(outpath, os.path.basename(line[1]))