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

oops

Browse Source
This commit is contained in:
mrq 2023-08-14 10:50:40 -05:00
parent 13571380be
commit 5fa86182b5
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data/config.yaml
View File

@ -99,7 +99,4 @@ inference:
use_vocos: True use_vocos: True
bitsandbytes: bitsandbytes:
enabled: false enabled: false
device: cuda
distributed: False

2
vall_e/data.py
View File

@ -313,7 +313,7 @@ def _create_dataloader(dataset, training):
return DataLoader( return DataLoader(
dataset=dataset, dataset=dataset,
batch_size=cfg.hyperparameters.batch_size if training else cfg.evaluation.batch_size, batch_size=cfg.hyperparameters.batch_size if training else cfg.evaluation.batch_size,
shuffle=False if cfg.distributed else True, # training shuffle=False, # if cfg.distributed else True, # training
drop_last=training, drop_last=training,
num_workers=cfg.dataset.workers, num_workers=cfg.dataset.workers,
collate_fn=collate_fn, collate_fn=collate_fn,