From f00c3acb5e1aafdb34cb478f0fb5206fd958819b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=93=9D=E6=A2=A6=E5=AE=9E?=
 <36986837+SapphireLab@users.noreply.github.com>
Date: Mon, 15 Jul 2024 13:19:51 +0800
Subject: [PATCH] fix parameters for fs2 (#225)

---
 models/tts/fastspeech2/fs2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/tts/fastspeech2/fs2.py b/models/tts/fastspeech2/fs2.py
index 464179f7..61cab7eb 100644
--- 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