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 remote-tracking branch 'origin/inference_streaming' into inference_streaming

Browse Source
This commit is contained in:
禾息
2024-09-03 11:13:25 +08:00
parent a801416805 d2dea3d928
commit 752103a307
15 changed files with 754 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/flow/flow_matching.py
View File

@@ -82,10 +82,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,