Flux9665 commited on
Commit
b88dcca
·
1 Parent(s): c2d9a9c

fix missing language id input

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: