From b10c58436d6871c26485d30b203e6cfdd4167602 Mon Sep 17 00:00:00 2001
From: mrq <mrq@ecker.tech>
Date: Sun, 20 Aug 2023 22:41:55 -0500
Subject: [PATCH] pesky dot
---
tortoise/utils/audio.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/tortoise/utils/audio.py b/tortoise/utils/audio.py
index 4302c40..3bb3a8d 100755
--- a/tortoise/utils/audio.py
+++ b/tortoise/utils/audio.py
@@ -101,7 +101,7 @@ def get_voice( name, dir=get_voice_dir(), load_latents=True, extensions=["wav",
files = os.listdir(subj)
if load_latents:
- extensions.append(".pth")
+ extensions.append("pth")
voice = []
for file in files:
@@ -113,7 +113,7 @@ def get_voice( name, dir=get_voice_dir(), load_latents=True, extensions=["wav",
return sorted( voice )
-def get_voice_list(dir=get_voice_dir(), append_defaults=False, extensions=["wav", "mp3", "flac", "pth"]):
+def get_voice_list(dir=get_voice_dir(), append_defaults=False, load_latents=True, extensions=["wav", "mp3", "flac"]):
defaults = [ "random", "microphone" ]
os.makedirs(dir, exist_ok=True)
#res = sorted([d for d in os.listdir(dir) if d not in defaults and os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 ])
@@ -126,7 +126,7 @@ def get_voice_list(dir=get_voice_dir(), append_defaults=False, extensions=["wav"
continue
if len(os.listdir(os.path.join(dir, name))) == 0:
continue
- files = get_voice( name, dir=dir, extensions=extensions )
+ files = get_voice( name, dir=dir, extensions=extensions, load_latents=load_latents )
if len(files) > 0:
res.append(name)
@@ -134,7 +134,7 @@ def get_voice_list(dir=get_voice_dir(), append_defaults=False, extensions=["wav"
for subdir in os.listdir(f'{dir}/{name}'):
if not os.path.isdir(f'{dir}/{name}/{subdir}'):
continue
- files = get_voice( f'{name}/{subdir}', dir=dir, extensions=extensions )
+ files = get_voice( f'{name}/{subdir}', dir=dir, extensions=extensions, load_latents=load_latents )
if len(files) == 0:
continue
res.append(f'{name}/{subdir}')