mrq/tortoise-tts
4
6
Fork 15
Code Issues 9 Pull Requests 1 Packages Projects Releases Wiki Activity

Merge pull request #90 from MarcusLlewellyn/read_combine

Browse Source 
read.py combines all candidates
This commit is contained in:
James Betker 2022-06-06 14:59:35 -06:00 committed by GitHub
commit 5d96b486fb
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tortoise/read.py
View File

@ -72,11 +72,22 @@ if __name__ == '__main__':
gen = gen[0].squeeze(0).cpu() gen = gen[0].squeeze(0).cpu()
all_parts.append(gen) all_parts.append(gen)
full_audio = torch.cat(all_parts, dim=-1) if args.candidates == 1:
torchaudio.save(os.path.join(voice_outpath, 'combined.wav'), full_audio, 24000) full_audio = torch.cat(all_parts, dim=-1)
torchaudio.save(os.path.join(voice_outpath, 'combined.wav'), full_audio, 24000)
if args.produce_debug_state: if args.produce_debug_state:
os.makedirs('debug_states', exist_ok=True) os.makedirs('debug_states', exist_ok=True)
dbg_state = (seed, texts, voice_samples, conditioning_latents) dbg_state = (seed, texts, voice_samples, conditioning_latents)
torch.save(dbg_state, f'debug_states/read_debug_{selected_voice}.pth') torch.save(dbg_state, f'debug_states/read_debug_{selected_voice}.pth')
# Combine each candidate's audio clips.
if args.candidates > 1:
audio_clips = []
for candidate in range(args.candidates):
for line in range(len(texts)):
wav_file = os.path.join(voice_outpath, str(line), f"{candidate}.wav")
audio_clips.append(load_audio(wav_file, 24000))
audio_clips = torch.cat(audio_clips, dim=-1)
torchaudio.save(os.path.join(voice_outpath, f"combined_{candidate:02d}.wav"), audio_clips, 24000)
audio_clips = []