From e0f9be76c0f7c4f1f7de372aff93ef0eb6b32e84 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Sun, 20 Feb 2022 11:36:27 +0100 Subject: [PATCH] Update test_run in wavernn and wavegrad --- TTS/vocoder/models/wavegrad.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/TTS/vocoder/models/wavegrad.py b/TTS/vocoder/models/wavegrad.py index 95aa3cd2..02c28c23 100644 --- a/TTS/vocoder/models/wavegrad.py +++ b/TTS/vocoder/models/wavegrad.py @@ -307,7 +307,9 @@ class Wavegrad(BaseVocoder): y = y.unsqueeze(1) return {"input": m, "waveform": y} - def get_data_loader(self, config: Coqpit, assets: Dict, is_eval: True, samples: List, verbose: bool, num_gpus: int): + def get_data_loader( + self, config: Coqpit, assets: Dict, is_eval: True, samples: List, verbose: bool, num_gpus: int + ): ap = assets["audio_processor"] dataset = WaveGradDataset( ap=ap,