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

fixed the overwrite regression for bark and vall-e backends too

Browse Source
This commit is contained in:
mrq 2023-08-24 19:46:42 +00:00
parent f5fab33e9c
commit 533b73e083
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

@ -432,7 +432,7 @@ def generate_bark(**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"^{cleanup_voice_name(voice)}_(\d+)(?:.+?)?{extension}$", filename) match = re.findall(rf"^{cleanup_voice_name(voice)}_(\d+)(?:.+?)?{extension}$", filename)
if match and len(match) > 0: if match and len(match) > 0:
@ -734,7 +734,7 @@ def generate_valle(**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: