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

Fix batch_group_size in VITS

This commit is contained in:
Eren Gölge 2022-05-07 13:40:34 +02:00
parent fa887ef5f9
commit 3f03e3012c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/models/vits.py
View File

@ -1452,7 +1452,7 @@ class Vits(BaseTTS):
# init dataloader # init dataloader
dataset = VitsDataset( dataset = VitsDataset(
samples=samples, samples=samples,
# batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size, batch_group_size=0 if is_eval else config.batch_group_size * config.batch_size,
min_text_len=config.min_text_len, min_text_len=config.min_text_len,
max_text_len=config.max_text_len, max_text_len=config.max_text_len,
min_audio_len=config.min_audio_len, min_audio_len=config.min_audio_len,