Update cosyvoice/cli/model.py

cosyvoice/cli/model.py

@@ -25,11 +25,11 @@ class CosyVoiceModel:
         self.hift = hift
 
     def load(self, llm_model, flow_model, hift_model):
-        self.llm.load_state_dict(torch.load(llm_model, map_location=self.device))
+        self.llm.load_state_dict(torch.load(llm_model, map_location="cpu"))
         self.llm.to(self.device).eval()
-        self.flow.load_state_dict(torch.load(flow_model, map_location=self.device))
+        self.flow.load_state_dict(torch.load(flow_model, map_location="cpu"))
         self.flow.to(self.device).eval()
-        self.hift.load_state_dict(torch.load(hift_model, map_location=self.device))
+        self.hift.load_state_dict(torch.load(hift_model, map_location="cpu"))
         self.hift.to(self.device).eval()
 
     def inference(self, text, text_len, flow_embedding, llm_embedding=torch.zeros(0, 192),