Flux9665 commited on
Commit
32ade73
β€’
1 Parent(s): f9b7eb9

Update InferenceInterfaces/Meta_FastSpeech2.py

Browse files
InferenceInterfaces/Meta_FastSpeech2.py CHANGED
@@ -17,7 +17,7 @@ class Meta_FastSpeech2(torch.nn.Module):
17
  def __init__(self, device="cpu"):
18
  super().__init__()
19
  model_name = "Meta"
20
- language = "en"
21
  self.device = device
22
  self.text2phone = ArticulatoryCombinedTextFrontend(language=language, add_silence_to_end=True)
23
  checkpoint = torch.load(os.path.join("Models", f"FastSpeech2_{model_name}", "best.pt"), map_location='cpu')
@@ -52,7 +52,8 @@ class Meta_FastSpeech2(torch.nn.Module):
52
  utterance_embedding=self.default_utterance_embedding,
53
  durations=durations,
54
  pitch=pitch,
55
- energy=energy)
 
56
  mel = mel.transpose(0, 1)
57
  wave = self.mel2wav(mel)
58
  if view:
 
17
  def __init__(self, device="cpu"):
18
  super().__init__()
19
  model_name = "Meta"
20
+ language = "de"
21
  self.device = device
22
  self.text2phone = ArticulatoryCombinedTextFrontend(language=language, add_silence_to_end=True)
23
  checkpoint = torch.load(os.path.join("Models", f"FastSpeech2_{model_name}", "best.pt"), map_location='cpu')
 
52
  utterance_embedding=self.default_utterance_embedding,
53
  durations=durations,
54
  pitch=pitch,
55
+ energy=energy,
56
+ lang_id=self.lang_id)
57
  mel = mel.transpose(0, 1)
58
  wave = self.mel2wav(mel)
59
  if view: