Flux9665 commited on
Commit
4ee44ad
1 Parent(s): 21e0fad

Update InferenceInterfaces/Meta_FastSpeech2.py

Browse files
InferenceInterfaces/Meta_FastSpeech2.py CHANGED
@@ -48,7 +48,8 @@ class Meta_FastSpeech2(torch.nn.Module):
48
  utterance_embedding=self.default_utterance_embedding,
49
  durations=durations,
50
  pitch=pitch,
51
- energy=energy)
 
52
  mel = mel.transpose(0, 1)
53
  wave = self.mel2wav(mel)
54
  if view:
 
48
  utterance_embedding=self.default_utterance_embedding,
49
  durations=durations,
50
  pitch=pitch,
51
+ energy=energy,
52
+ lang_id=self.lang_id)
53
  mel = mel.transpose(0, 1)
54
  wave = self.mel2wav(mel)
55
  if view: