Sfoglia il codice sorgente

Merge pull request #603 from oongrider/f_free_cache_dict

release flow_cache_dict to prevent potential memory leaks in long-running processes
Xiang Lyu 1 anno fa
parent
commit
3914b54c82
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      cosyvoice/cli/model.py

+ 1 - 0
cosyvoice/cli/model.py

@@ -201,6 +201,7 @@ class CosyVoiceModel:
             self.llm_end_dict.pop(this_uuid)
             self.mel_overlap_dict.pop(this_uuid)
             self.hift_cache_dict.pop(this_uuid)
+            self.flow_cache_dict.pop(this_uuid)
 
     def vc(self, source_speech_token, flow_prompt_speech_token, prompt_speech_feat, flow_embedding, stream=False, speed=1.0, **kwargs):
         # this_uuid is used to track variables related to this inference thread