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

add empty cache

Browse Source
This commit is contained in:
lyuxiang.lx
2024-07-09 23:48:23 +08:00
parent 7981796523
commit ee9e87b4d3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
cosyvoice/cli/model.py
View File

@@ -56,4 +56,5 @@ class CosyVoiceModel:
prompt_feat_len=prompt_speech_feat_len.to(self.device), prompt_feat_len=prompt_speech_feat_len.to(self.device),
embedding=flow_embedding.to(self.device)) embedding=flow_embedding.to(self.device))
tts_speech = self.hift.inference(mel=tts_mel).cpu() tts_speech = self.hift.inference(mel=tts_mel).cpu()
torch.cuda.empty_cache()
return {'tts_speech': tts_speech} return {'tts_speech': tts_speech}