Skip to content

GitLab

D
dhruva-iitm-tts
Commit 7d911f76 authored by Nikhilesh Bhatnagar's avatar Nikhilesh Bhatnagar
Browse files
Options

Forgot to add female voices

parent 52a974e2
Pipeline #29 failed with stages
      in 0 seconds
      Hide whitespace changes
      Inline Side-by-side
      Showing with 3 additions and 3 deletions
      triton_models/tts/1/model.py
      View file @ 7d911f76
      ...@@ -59,18 +59,18 @@ class TritonPythonModel: ...@@ -59,18 +59,18 @@ class TritonPythonModel:
      else: else:
      self.preprocessors[lang] = self.tts_dur_align_preprocessor self.preprocessors[lang] = self.tts_dur_align_preprocessor
      self.models = {} self.models = {}
      for (language, _), gender in product(LANGMAP.values(), ("male",)): for (language, _), gender in product(LANGMAP.values(), ("male", "female")):
      try: try:
      self.models[(language, gender)] = self.load_fastspeech2_model( self.models[(language, gender)] = self.load_fastspeech2_model(
      language, language,
      gender, gender,
      f"cuda", "cuda",
      ) )
      except: except:
      pass pass
      self.vocoders = { self.vocoders = {
      (gender, family): self.load_vocoder(gender, family, "cuda") (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): 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