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

update vc code

Browse Source
This commit is contained in:
lyuxiang.lx
2024-09-26 14:46:24 +08:00
parent d52358f6c5
commit 06934c38c7
4 changed files with 11 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
cosyvoice/cli/model.py
View File

@@ -54,10 +54,9 @@ class CosyVoiceModel:
self.hift_cache_dict = {}
def load(self, llm_model, flow_model, hift_model):
if self.llm is not None:
self.llm.load_state_dict(torch.load(llm_model, map_location=self.device))
self.llm.to(self.device).eval()
self.llm.half()
self.llm.load_state_dict(torch.load(llm_model, map_location=self.device))
self.llm.to(self.device).eval()
self.llm.half()
self.flow.load_state_dict(torch.load(flow_model, map_location=self.device))
self.flow.to(self.device).eval()
self.hift.load_state_dict(torch.load(hift_model, map_location=self.device))