fix filename generation which didn't work and overwrote existing files #341

Merged
mrq merged 1 commits from ben_mkiv/ai-voice-cloning:master into master 2023-08-24 12:30:01 +00:00

View File

@ -1099,7 +1099,7 @@ def generate_tortoise(**kwargs):
for i, file in enumerate(os.listdir(outdir)): for i, file in enumerate(os.listdir(outdir)):
filename = os.path.basename(file) filename = os.path.basename(file)
extension = os.path.splitext(filename)[-1][1:] extension = os.path.splitext(filename)[-1][1:]
if extension != ".json" and extension != ".wav": if extension != "json" and extension != "wav":
continue continue
match = re.findall(rf"^{voice}_(\d+)(?:.+?)?{extension}$", filename) match = re.findall(rf"^{voice}_(\d+)(?:.+?)?{extension}$", filename)
if match and len(match) > 0: if match and len(match) > 0: