terminator/ai-voice-cloning-terminator
1
1
Fork 0
forked from mrq/ai-voice-cloning
Code Pull Requests Activity

fix sloppy copy paste job when looking for new models

Browse Source
This commit is contained in:
mrq 2023-02-18 19:46:26 +00:00
parent ebbc85fb6a
commit cd8919e65c
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils.py
View File

@ -265,8 +265,6 @@ def generate(
keys = sorted(list(idx_cache.keys()))
idx = keys[-1] + 1
# I know there's something to pad I don't care
idx = pad(idx, 4)
def get_name(line=0, candidate=0, combined=False):
@ -724,7 +722,7 @@ def get_voice_list(dir=get_voice_dir()):
def get_autoregressive_models(dir="./models/finetunes/"):
os.makedirs(dir, exist_ok=True)
return [get_model_path('autoregressive.pth')] + sorted([d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 ])
return [get_model_path('autoregressive.pth')] + sorted([f'{dir}/{d}' for d in os.listdir(dir) if d[-4:] == ".pth" ])
def get_dataset_list(dir="./training/"):
return sorted([d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d)) and len(os.listdir(os.path.join(dir, d))) > 0 and "train.txt" in os.listdir(os.path.join(dir, d)) ])