neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-03-03 12:04:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix memoery leak in read.py

Browse source
This commit is contained in:
manmay nakhashi 2023-07-29 05:51:56 +05:30 committed by GitHub
parent 61e5d993b1
commit 5413ce57d5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tortoise/read.py
View file

@ -64,15 +64,15 @@ if __name__ == '__main__':
gen = tts.tts_with_preset(text, voice_samples=voice_samples, conditioning_latents=conditioning_latents,
preset=args.preset, k=args.candidates, use_deterministic_seed=seed)
if args.candidates == 1:
gen = gen.squeeze(0).cpu()
torchaudio.save(os.path.join(voice_outpath, f'{j}.wav'), gen, 24000)
audio_ = gen.squeeze(0).cpu()
torchaudio.save(os.path.join(voice_outpath, f'{j}.wav'), audio_, 24000)
else:
candidate_dir = os.path.join(voice_outpath, str(j))
os.makedirs(candidate_dir, exist_ok=True)
for k, g in enumerate(gen):
torchaudio.save(os.path.join(candidate_dir, f'{k}.wav'), g.squeeze(0).cpu(), 24000)
gen = gen[0].squeeze(0).cpu()
all_parts.append(gen)
audio_ = gen[0].squeeze(0).cpu()
all_parts.append(audio_)
if args.candidates == 1:
full_audio = torch.cat(all_parts, dim=-1)