Flux9665 commited on
Commit
83eb0e7
1 Parent(s): 787d1da

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