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

fix lint

Browse Source
This commit is contained in:
lyuxiang.lx
2025-06-29 13:21:44 +08:00
parent 46dfe0439b
commit 21a5efd8ae
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cosyvoice/llm/llm.py
View File

@@ -381,7 +381,7 @@ class Qwen2LM(TransformerLM):
self, self,
batch: dict, batch: dict,
device: torch.device, device: torch.device,
) -> Dict[str, Optional[torch.Tensor]]: ) -> Dict[str, Optional[torch.Tensor]]:
text_token = batch['text_token'].to(device) text_token = batch['text_token'].to(device)
text_token_len = batch['text_token_len'].to(device) text_token_len = batch['text_token_len'].to(device)
speech_token = batch['speech_token'].to(device) speech_token = batch['speech_token'].to(device)