mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

fall back to normal behavior if theres actually no audiofiles loaded from the dataset when using it for computing latents

Browse Source
This commit is contained in:
mrq 2023-03-11 16:46:03 +00:00
parent b90c164778
commit cf41492f76
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/utils.py
View File

@ -543,7 +543,7 @@ def compute_latents(voice=None, voice_samples=None, voice_latents_chunks=0, prog
with open(dataset_path, 'r', encoding="utf-8") as f: with open(dataset_path, 'r', encoding="utf-8") as f:
lines = f.readlines() lines = f.readlines()
print("Leveraging LJSpeech dataset for computing latents") print("Leveraging dataset for computing latents")
voice_samples = [] voice_samples = []
max_length = 0 max_length = 0
@ -558,6 +558,9 @@ def compute_latents(voice=None, voice_samples=None, voice_latents_chunks=0, prog
voice_samples[i] = pad_or_truncate(voice_samples[i], max_length) voice_samples[i] = pad_or_truncate(voice_samples[i], max_length)
voice_latents_chunks = len(voice_samples) voice_latents_chunks = len(voice_samples)
if voice_latents_chunks == 0:
print("Dataset is empty!")
load_from_dataset = True
if not load_from_dataset: if not load_from_dataset:
voice_samples, _ = load_voice(voice, load_latents=False) voice_samples, _ = load_voice(voice, load_latents=False)