From 533b73e083f846e57ac99386f361238210890d66 Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 24 Aug 2023 19:46:42 +0000 Subject: [PATCH] fixed the overwrite regression for bark and vall-e backends too --- src/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/utils.py b/src/utils.py index bbc7b0c..df6bf11 100755 --- a/src/utils.py +++ b/src/utils.py @@ -432,7 +432,7 @@ def generate_bark(**kwargs): for i, file in enumerate(os.listdir(outdir)): filename = os.path.basename(file) extension = os.path.splitext(filename)[-1][1:] - if extension != ".json" and extension != ".wav": + if extension != "json" and extension != "wav": continue match = re.findall(rf"^{cleanup_voice_name(voice)}_(\d+)(?:.+?)?{extension}$", filename) if match and len(match) > 0: @@ -734,7 +734,7 @@ def generate_valle(**kwargs): for i, file in enumerate(os.listdir(outdir)): filename = os.path.basename(file) extension = os.path.splitext(filename)[-1][1:] - if extension != ".json" and extension != ".wav": + if extension != "json" and extension != "wav": continue match = re.findall(rf"^{voice}_(\d+)(?:.+?)?{extension}$", filename) if match and len(match) > 0: