diff --git a/data/config.yaml b/data/config.yaml index a483998..2fd964a 100755 --- a/data/config.yaml +++ b/data/config.yaml @@ -99,7 +99,4 @@ inference: use_vocos: True bitsandbytes: - enabled: false - -device: cuda -distributed: False \ No newline at end of file + enabled: false \ No newline at end of file diff --git a/vall_e/data.py b/vall_e/data.py index 57a9c99..4828dca 100755 --- a/vall_e/data.py +++ b/vall_e/data.py @@ -313,7 +313,7 @@ def _create_dataloader(dataset, training): return DataLoader( dataset=dataset, 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, num_workers=cfg.dataset.workers, collate_fn=collate_fn,