diff --git a/triton_models/tts/1/model.py b/triton_models/tts/1/model.py index 65cd92918065ebfb1b59f890089d4a3e34d2c1e0..66e71bd7fb6a85e3485d188ce44d1a64da4df490 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):