NtTestAlert/tortoise-tts
1
0
Fork 0
forked from mrq/tortoise-tts
Code Issues Pull Requests Packages Projects Releases Wiki Activity

avoid mutable default in aligner

Browse Source
This commit is contained in:
Josh Ziegler 2022-05-26 16:20:09 -04:00
parent 7becd30c2a
commit 53f6563e3e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tortoise/utils/wav2vec_alignment.py
View File

@ -7,13 +7,15 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2FeatureExtractor, Wav2Vec2CTCTo
from tortoise.utils.audio import load_audio from tortoise.utils.audio import load_audio
def max_alignment(s1, s2, skip_character='~', record={}): def max_alignment(s1, s2, skip_character='~', record=None):
""" """
A clever function that aligns s1 to s2 as best it can. Wherever a character from s1 is not found in s2, a '~' is A clever function that aligns s1 to s2 as best it can. Wherever a character from s1 is not found in s2, a '~' is
used to replace that character. used to replace that character.
Finally got to use my DP skills! Finally got to use my DP skills!
""" """
if record is None:
record = {}
assert skip_character not in s1, f"Found the skip character {skip_character} in the provided string, {s1}" assert skip_character not in s1, f"Found the skip character {skip_character} in the provided string, {s1}"
if len(s1) == 0: if len(s1) == 0:
return '' return ''
@ -145,4 +147,3 @@ class Wav2VecAlignment:
start, stop = nri start, stop = nri
output_audio.append(audio[:, alignments[start]:alignments[stop]]) output_audio.append(audio[:, alignments[start]:alignments[stop]])
return torch.cat(output_audio, dim=-1) return torch.cat(output_audio, dim=-1)