Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update test_run in wavernn and wavegrad

This commit is contained in:
Eren Gölge 2022-02-20 11:36:27 +01:00
parent bf540f4323
commit e0f9be76c0
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/vocoder/models/wavegrad.py
View File

@ -307,7 +307,9 @@ class Wavegrad(BaseVocoder):
y = y.unsqueeze(1) y = y.unsqueeze(1)
return {"input": m, "waveform": y} 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"] ap = assets["audio_processor"]
dataset = WaveGradDataset( dataset = WaveGradDataset(
ap=ap, ap=ap,