From be2ad6ea8534cca9a654376d28cc4d1fb9b2b3f1 Mon Sep 17 00:00:00 2001 From: Aidan Pine Date: Thu, 3 Jun 2021 09:26:03 -0700 Subject: [PATCH] Update modules.py --- model/modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/modules.py b/model/modules.py index f694d56a6e..01450bbabc 100644 --- a/model/modules.py +++ b/model/modules.py @@ -143,7 +143,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