mirror of
https://github.com/FunAudioLLM/CosyVoice.git
synced 2026-02-05 01:49:25 +08:00
add empty cache
This commit is contained in:
@@ -56,4 +56,5 @@ class CosyVoiceModel:
|
||||
prompt_feat_len=prompt_speech_feat_len.to(self.device),
|
||||
embedding=flow_embedding.to(self.device))
|
||||
tts_speech = self.hift.inference(mel=tts_mel).cpu()
|
||||
torch.cuda.empty_cache()
|
||||
return {'tts_speech': tts_speech}
|
||||
|
||||
Reference in New Issue
Block a user