Spaces:
Running
on
Zero
Running
on
Zero
Update Modules/ToucanTTS/InferenceToucanTTS.py
Browse files
Modules/ToucanTTS/InferenceToucanTTS.py
CHANGED
@@ -216,6 +216,8 @@ class ToucanTTS(torch.nn.Module):
|
|
216 |
|
217 |
# encoding the texts
|
218 |
text_masks = make_non_pad_mask(text_lengths, device=text_lengths.device).unsqueeze(-2)
|
|
|
|
|
219 |
encoded_texts, _ = self.encoder(text_tensors, text_masks, utterance_embedding=utterance_embedding, lang_ids=lang_ids)
|
220 |
|
221 |
# predicting pitch, energy and durations
|
|
|
216 |
|
217 |
# encoding the texts
|
218 |
text_masks = make_non_pad_mask(text_lengths, device=text_lengths.device).unsqueeze(-2)
|
219 |
+
print(text_tensors)
|
220 |
+
print(text_masks)
|
221 |
encoded_texts, _ = self.encoder(text_tensors, text_masks, utterance_embedding=utterance_embedding, lang_ids=lang_ids)
|
222 |
|
223 |
# predicting pitch, energy and durations
|