deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-04 17:39:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
lyuxiang.lx
2025-02-11 00:07:12 +08:00
parent b6d66ce2e3
commit f77c6a85aa
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/cli/model.py
View File

@@ -398,6 +398,10 @@ class CosyVoice2Model(CosyVoiceModel):
prompt_speech_feat=torch.zeros(1, 0, 80), stream=False, speed=1.0, **kwargs):
# this_uuid is used to track variables related to this inference thread
this_uuid = str(uuid.uuid1())
# NOTE flow model is only trained with static_chunk_size, so we need to trim flow prompt
n_chunk = int(flow_prompt_speech_token.size(1) / self.token_hop_len)
flow_prompt_speech_token = flow_prompt_speech_token[:, :n_chunk * self.token_hop_len]
prompt_speech_feat = prompt_speech_feat[:, :n_chunk * self.token_hop_len * 2]
with self.lock:
self.tts_speech_token_dict[this_uuid], self.llm_end_dict[this_uuid] = [], False
self.hift_cache_dict[this_uuid] = None