2 anni fa · ad6fd717ab
--- a/fish_speech/configs/vqgan_pretrain_v2.yaml
+++ b/fish_speech/configs/vqgan_pretrain_v2.yaml
@@ -7,7 +7,7 @@ project: vqgan_pretrain_v2
 
															 # Lightning Trainer
														
 
															 trainer:
														
 
															   accelerator: gpu
														
 
															-  devices: 4
														
 
															+  devices: auto
														
 
															   strategy: ddp_find_unused_parameters_true
														
 
															   precision: 32
														
 
															   max_steps: 1_000_000