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

oops (fixed proms being erased from a config oversight)

Browse Source
This commit is contained in:
mrq 2024-07-25 12:39:57 -05:00
parent 1acb0e9c84
commit 682e4387dc
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vall_e/config.py
View File

@ -152,7 +152,7 @@ class Dataset:
random_utterance: float = 1.0 random_utterance: float = 1.0
max_prompts: int = 3 max_prompts: int = 3
prompt_duration: float = 0.0 # legacy prompt_duration: float | None = None # legacy
max_resps: int = 1 max_resps: int = 1
p_resp_append: float = 1.0 p_resp_append: float = 1.0
@ -839,6 +839,7 @@ class Config(BaseConfig):
if self.hyperparameters.scheduler == "": if self.hyperparameters.scheduler == "":
self.hyperparameters.torch_scheduler = True self.hyperparameters.torch_scheduler = True
if self.dataset.prompt_duration is not None:
self.dataset.prompt_duration_range = [self.dataset.prompt_duration, self.dataset.prompt_duration] self.dataset.prompt_duration_range = [self.dataset.prompt_duration, self.dataset.prompt_duration]
if self.trainer.backend == "local" and self.distributed: if self.trainer.backend == "local" and self.distributed:

2
vall_e/data.py
View File

@ -434,7 +434,7 @@ class Dataset(_Dataset):
self.training = training self.training = training
self.dataset_type = "training" if self.training else "validation" self.dataset_type = "training" if self.training else "validation"
self.dataset = cfg.dataset.training if self.training else cfg.dataset.validation self.dataset = cfg.dataset.training if self.training else cfg.dataset.validation
self.sampler_type = cfg.dataset.sample_type # if self.dataset_type == "training" else "group" self.sampler_type = cfg.dataset.sample_type if self.dataset_type == "training" else "path"
self.sampler_order = cfg.dataset.sample_order self.sampler_order = cfg.dataset.sample_order
self.sampler_shuffle = cfg.dataset.sample_shuffle self.sampler_shuffle = cfg.dataset.sample_shuffle