Commit 7d911f76 authored by Nikhilesh Bhatnagar's avatar Nikhilesh Bhatnagar

Forgot to add female voices

parent 52a974e2
Pipeline #29 failed with stages
in 0 seconds
......@@ -59,18 +59,18 @@ class TritonPythonModel:
else:
self.preprocessors[lang] = self.tts_dur_align_preprocessor
self.models = {}
for (language, _), gender in product(LANGMAP.values(), ("male",)):
for (language, _), gender in product(LANGMAP.values(), ("male", "female")):
try:
self.models[(language, gender)] = self.load_fastspeech2_model(
language,
gender,
f"cuda",
"cuda",
)
except:
pass
self.vocoders = {
(gender, family): self.load_vocoder(gender, family, "cuda")
for gender, family in product(("male",), ("aryan", "dravidian"))
for gender, family in product(("male", "female"), ("aryan", "dravidian"))
}
def load_vocoder(self, gender, family, device):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment