From ead9644e861cedd28e5ae27ab2df28e24222e6ad Mon Sep 17 00:00:00 2001
From: "lyuxiang.lx" <lyuxiang.lx@alibaba-inc.com>
Date: Thu, 18 Jul 2024 11:18:45 +0800
Subject: [PATCH] change flow config in cosyvoice.fromscratch.yaml

---
 .../libritts/cosyvoice/conf/cosyvoice.fromscratch.yaml    | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/libritts/cosyvoice/conf/cosyvoice.fromscratch.yaml b/examples/libritts/cosyvoice/conf/cosyvoice.fromscratch.yaml
index 30545ff..54b6e7b 100644
--- a/examples/libritts/cosyvoice/conf/cosyvoice.fromscratch.yaml
+++ b/examples/libritts/cosyvoice/conf/cosyvoice.fromscratch.yaml
@@ -65,9 +65,9 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec
     only_mask_loss: True
     encoder: !new:cosyvoice.transformer.encoder.ConformerEncoder
         output_size: 512
-        attention_heads: 8
-        linear_units: 2048
-        num_blocks: 6
+        attention_heads: 4
+        linear_units: 1024
+        num_blocks: 3
         dropout_rate: 0.1
         positional_dropout_rate: 0.1
         attention_dropout_rate: 0.1
@@ -100,7 +100,7 @@ flow: !new:cosyvoice.flow.flow.MaskedDiffWithXvec
             dropout: 0
             attention_head_dim: 64
             n_blocks: 4
-            num_mid_blocks: 12
+            num_mid_blocks: 8
             num_heads: 8
             act_fn: 'gelu'