mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

require minimum of 1 second durations for training because of my slop code auto-transposing that I don't wanna fix right now

Browse Source
This commit is contained in:
mrq 2025-02-26 22:00:33 -06:00
parent cbd4d7d7f4
commit 06ef3daf3c
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vall_e/data.py
View File

@ -707,6 +707,8 @@ def _get_duration_map( type="training" ):
return _durations_map[type] if type in _durations_map else {} return _durations_map[type] if type in _durations_map else {}
def _load_paths(dataset, type="training", silent=not is_global_leader(), dataset_hash_key=None): def _load_paths(dataset, type="training", silent=not is_global_leader(), dataset_hash_key=None):
assert cfg.dataset.min_duration >= 1.0, "Minimum duration too low."
if not dataset_hash_key: if not dataset_hash_key:
dataset_hash_key = cfg.dataset.hash_key(sorted(dataset)) dataset_hash_key = cfg.dataset.hash_key(sorted(dataset))
@ -862,6 +864,7 @@ class Dataset(_Dataset):
extra_paths_by_spkr_name: dict[str, list] = {}, extra_paths_by_spkr_name: dict[str, list] = {},
): ):
super().__init__() super().__init__()
self._head = None self._head = None
self.sampler = None self.sampler = None

2
vall_e/train.py
View File

@ -35,7 +35,7 @@ def train_feeder(engine, batch, teacher=None):
engine.current_batch_size = batch_size engine.current_batch_size = batch_size
output = engine( output = engine(
phn_list=batch["phns"], phns_list=batch["phns"],
proms_list=batch["proms"], proms_list=batch["proms"],
resps_list=batch["resps"], resps_list=batch["resps"],
lang_list=batch["lang"], lang_list=batch["lang"],