From 7d911f7627a30f5b11b9d9bc9c5858752b5999a2 Mon Sep 17 00:00:00 2001 From: Nikhilesh Bhatnagar Date: Mon, 30 Oct 2023 13:22:47 +0000 Subject: [PATCH] Forgot to add female voices --- triton_models/tts/1/model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/triton_models/tts/1/model.py b/triton_models/tts/1/model.py index 65cd929..66e71bd 100644 --- a/triton_models/tts/1/model.py +++ b/triton_models/tts/1/model.py @@ -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): -- GitLab