mirror of
https://github.com/FunAudioLLM/CosyVoice.git
synced 2026-02-05 18:09:24 +08:00
change flow config in cosyvoice.fromscratch.yaml
This commit is contained in:
@@ -65,9 +65,9 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec
|
|||||||
only_mask_loss: True
|
only_mask_loss: True
|
||||||
encoder: !new:cosyvoice.transformer.encoder.ConformerEncoder
|
encoder: !new:cosyvoice.transformer.encoder.ConformerEncoder
|
||||||
output_size: 512
|
output_size: 512
|
||||||
attention_heads: 8
|
attention_heads: 4
|
||||||
linear_units: 2048
|
linear_units: 1024
|
||||||
num_blocks: 6
|
num_blocks: 3
|
||||||
dropout_rate: 0.1
|
dropout_rate: 0.1
|
||||||
positional_dropout_rate: 0.1
|
positional_dropout_rate: 0.1
|
||||||
attention_dropout_rate: 0.1
|
attention_dropout_rate: 0.1
|
||||||
@@ -100,7 +100,7 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec
|
|||||||
dropout: 0
|
dropout: 0
|
||||||
attention_head_dim: 64
|
attention_head_dim: 64
|
||||||
n_blocks: 4
|
n_blocks: 4
|
||||||
num_mid_blocks: 12
|
num_mid_blocks: 8
|
||||||
num_heads: 8
|
num_heads: 8
|
||||||
act_fn: 'gelu'
|
act_fn: 'gelu'
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user