yasserrmd commited on
Commit
0ed5781
·
verified ·
1 Parent(s): 4a46ced

Update generate_audio.py

Browse files
Files changed (1) hide show
  1. generate_audio.py +1 -1
generate_audio.py CHANGED
@@ -90,7 +90,7 @@ class TTSGenerator:
90
  int: Sampling rate.
91
  """
92
 
93
- input_ids = self.parler_tokenizer(self.speaker2_description, return_tensors="pt").input_ids.to(self.device)
94
  prompt_input_ids = self.parler_tokenizer(text, return_tensors="pt").input_ids.to(self.device)
95
  generation = self.parler_model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
96
  audio_arr = generation.cpu().numpy().squeeze()
 
90
  int: Sampling rate.
91
  """
92
 
93
+ input_ids = self.parler_tokenizer(self.speaker2_description, return_tensors="pt").input_ids.to(self.device)
94
  prompt_input_ids = self.parler_tokenizer(text, return_tensors="pt").input_ids.to(self.device)
95
  generation = self.parler_model.generate(input_ids=input_ids, prompt_input_ids=prompt_input_ids)
96
  audio_arr = generation.cpu().numpy().squeeze()