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

oops

Browse Source
This commit is contained in:
mrq 2023-03-08 16:09:29 +00:00
parent 40e8d0774e
commit 0ab091e7ff
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils.py
View File

@ -297,14 +297,14 @@ def generate(
latents_path = f'{get_voice_dir()}/{voice}/cond_latents.pth' latents_path = f'{get_voice_dir()}/{voice}/cond_latents.pth'
if voice == "random" or voice == "microphone": if voice == "random" or voice == "microphone":
if latents and settings['conditioning_latents']: if latents and settings is not None and settings['conditioning_latents']:
dir = f'{get_voice_dir()}/{voice}/' dir = f'{get_voice_dir()}/{voice}/'
if not os.path.isdir(dir): if not os.path.isdir(dir):
os.makedirs(dir, exist_ok=True) os.makedirs(dir, exist_ok=True)
latents_path = f'{dir}/cond_latents.pth' latents_path = f'{dir}/cond_latents.pth'
torch.save(conditioning_latents, latents_path) torch.save(conditioning_latents, latents_path)
else: else:
if settings and "model_hash" in settings: if settings is not None and "model_hash" in settings:
latents_path = f'{get_voice_dir()}/{voice}/cond_latents_{settings["model_hash"][:8]}.pth' latents_path = f'{get_voice_dir()}/{voice}/cond_latents_{settings["model_hash"][:8]}.pth'
else: else:
latents_path = f'{get_voice_dir()}/{voice}/cond_latents_{tts.autoregressive_model_hash[:8]}.pth' latents_path = f'{get_voice_dir()}/{voice}/cond_latents_{tts.autoregressive_model_hash[:8]}.pth'