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()
|