Hugging Face's logo

Spaces:
kevinwang676
/
CosyVoice-hf
Runtime error

App Files Community
kevinwang676 commited on
Commit
985277a
·
verified ·
1 Parent(s): f4aa0cf

Update cosyvoice/cli/model.py

Browse files
Files changed (1) hide show
  1. cosyvoice/cli/model.py +1 -1
cosyvoice/cli/model.py CHANGED
@@ -27,7 +27,7 @@ class CosyVoiceModel:
27
  def load(self, llm_model, flow_model, hift_model):
28
  self.llm.load_state_dict(torch.load(llm_model, map_location="cpu"))
29
  self.llm.to(self.device).eval()
30
- self.flow.load_state_dict(torch.load(flow_model, map_location=self.device))
31
  self.flow.to(self.device).eval()
32
  self.hift.load_state_dict(torch.load(hift_model, map_location="cpu"))
33
  self.hift.to(self.device).eval()
 
27
  def load(self, llm_model, flow_model, hift_model):
28
  self.llm.load_state_dict(torch.load(llm_model, map_location="cpu"))
29
  self.llm.to(self.device).eval()
30
+ self.flow.load_state_dict(torch.load(flow_model, map_location="cpu"))
31
  self.flow.to(self.device).eval()
32
  self.hift.load_state_dict(torch.load(hift_model, map_location="cpu"))
33
  self.hift.to(self.device).eval()