Spaces:
Runtime error
Runtime error
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 = "
|
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:
|