From 602ca85c7434f67b9c1f27ea6bfeecb25f06fe3f Mon Sep 17 00:00:00 2001
From: WeberJulian <julian.weber@hotmail.fr>
Date: Tue, 28 Jun 2022 14:37:54 +0200
Subject: [PATCH] Fix pred_avg_pitch_emb

---
 TTS/tts/models/vits.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py
index 208a76d5..98beb97a 100644
--- a/TTS/tts/models/vits.py
+++ b/TTS/tts/models/vits.py
@@ -1727,7 +1727,8 @@ class Vits(BaseTTS):
 
         attn_mask = x_mask * y_mask.transpose(1, 2)  # [B, 1, T_enc] * [B, T_dec, 1]
         attn = generate_path(w_ceil.squeeze(1), attn_mask.squeeze(1).transpose(1, 2))
-
+        
+        pred_avg_pitch_emb = None
         if self.args.use_pitch:
             _, _, pred_avg_pitch_emb = self.forward_pitch_predictor(x, x_lengths, g_pp=g, pitch_transform=pitch_transform)
             x = x + pred_avg_pitch_emb