deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 18:09:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'inference_streaming' into flow_tensorrt

Browse Source
This commit is contained in:
Xiang Lyu
2024-08-30 14:20:06 +08:00
committed by GitHub
parent 18599be8d5 662012999a
commit ee988420f3
16 changed files with 754 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/flow/flow_matching.py
View File

@@ -78,10 +78,10 @@ class ConditionalCFM(BASECFM):
sol = []
for step in range(1, len(t_span)):
dphi_dt = self.forward_estimator(x, mask, mu, t, spks, cond)
dphi_dt = self.estimator(x, mask, mu, t, spks, cond)
# Classifier-Free Guidance inference introduced in VoiceBox
if self.inference_cfg_rate > 0:
cfg_dphi_dt = self.forward_estimator(
cfg_dphi_dt = self.estimator(
x, mask,
torch.zeros_like(mu), t,
torch.zeros_like(spks) if spks is not None else None,