From a0547bb383448e424a47248d9682a561ec28507f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Mon, 21 Aug 2023 07:25:24 +0800
Subject: [PATCH] Update data_utils.py

---
 data_utils.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/data_utils.py b/data_utils.py
index 4dd5232..7285672 100644
--- a/data_utils.py
+++ b/data_utils.py
@@ -30,6 +30,10 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
         self.sampling_rate = hparams.sampling_rate
         self.spk_map = hparams.spk2id
 
+        self.use_mel_spec_posterior = getattr(hparams, "use_mel_posterior_encoder", False)
+        if self.use_mel_spec_posterior:
+            self.n_mel_channels = getattr(hparams, "n_mel_channels", 80)
+
         self.cleaned_text = getattr(hparams, "cleaned_text", False)
 
         self.add_blank = hparams.add_blank