fix parameters for fs2 (#225)

open-mmlab · Jul 15, 2024 · f00c3ac · f00c3ac
1 parent 65ddb1e
commit f00c3ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/tts/fastspeech2/fs2.py b/models/tts/fastspeech2/fs2.py
@@ -215,7 +215,7 @@ def forward(
             x = x + pitch_embedding
         if self.energy_feature_level == "frame_level":
             energy_prediction, energy_embedding = self.get_energy_embedding(
-                x, energy_target, mel_mask, p_control
+                x, energy_target, mel_mask, e_control
             )
             x = x + energy_embedding