Spaces:
Running
Running
Update generate_audio.py
Browse files- generate_audio.py +1 -1
generate_audio.py
CHANGED
@@ -90,7 +90,7 @@ class TTSGenerator:
|
|
90 |
int: Sampling rate.
|
91 |
"""
|
92 |
|
93 |
-
|
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()
|