Lengyue 2 anni fa
parent
commit
e4839d6e84
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      fish_speech/configs/llama_finetune.yaml

+ 1 - 1
fish_speech/configs/llama_finetune.yaml

@@ -41,7 +41,7 @@ data:
   _target_: fish_speech.datasets.text.TextDataModule
   _target_: fish_speech.datasets.text.TextDataModule
   train_dataset: ${train_dataset}
   train_dataset: ${train_dataset}
   val_dataset: ${val_dataset}
   val_dataset: ${val_dataset}
-  num_workers: 1
+  num_workers: 4
   batch_size: 8
   batch_size: 8
   tokenizer: ${tokenizer}
   tokenizer: ${tokenizer}