flow-matching步数参数开放。

duj12 · Dec 4, 2024 · a765873 · a765873
1 parent 145d850
commit a765873
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/cosyvoice/flow/flow.py b/cosyvoice/flow/flow.py
@@ -111,7 +111,8 @@ def inference(self,
                   prompt_feat_len,
                   embedding,
                   flow_cache,
-                  sample_rate,):
+                  sample_rate,
+                  n_timesteps=10):
         assert token.shape[0] == 1
 
         # xvec projection
@@ -141,7 +142,7 @@ def inference(self,
             mask=mask.unsqueeze(1),
             spks=embedding,
             cond=conds,
-            n_timesteps=10,
+            n_timesteps=n_timesteps,
             prompt_len=mel_len1,
             flow_cache=flow_cache
         )

diff --git a/examples/tts_vc/cosyvoice/run_llm_phoneme.sh b/examples/tts_vc/cosyvoice/run_llm_phoneme.sh
@@ -12,7 +12,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3,4,5,6,7"
 num_gpus=$(echo $CUDA_VISIBLE_DEVICES | awk -F "," '{print NF}')
 job_id=1986
 dist_backend="nccl"
-num_workers=4
+num_workers=8
 prefetch=100
 train_engine=torch_ddp
 if [ ${stage} -le 0 ] && [ ${stop_stage} -ge 0 ]; then