Lengyue 2 anni fa
parent
commit
7c3c493fba

+ 1 - 2
fish_speech/configs/vqgan_finetune.yaml

@@ -50,8 +50,7 @@ model:
   sample_rate: ${sample_rate}
   hop_length: ${hop_length}
   segment_size: 8192
-  freeze_hifigan: false
-  freeze_vq: true
+  mode: finetune
 
   downsample:
     _target_: fish_speech.models.vqgan.modules.encoders.ConvDownSampler

+ 1 - 1
fish_speech/configs/vqgan_pretrain.yaml

@@ -48,7 +48,7 @@ model:
   sample_rate: ${sample_rate}
   hop_length: ${hop_length}
   segment_size: 8192
-  freeze_hifigan: false
+  mode: pretrain-stage1
 
   downsample:
     _target_: fish_speech.models.vqgan.modules.encoders.ConvDownSampler