开始部署

tts/cosyvoice/cli/model.py

@@ -76,7 +76,7 @@ class CosyVoiceModel:
             self.llm.load_state_dict(torch.load(llm_model, map_location="cpu"), strict=True)
         self.flow.load_state_dict(torch.load(flow_model,map_location="cpu"), strict=True)
         self.flow.eval()
-        hift_state_dict = {k.replace('generator.', ''): v for k, v in torch.load(hift_model, map_location=self.device).items()}
+        hift_state_dict = {k.replace('generator.', ''): v for k, v in torch.load(hift_model, map_location="cpu").items()}
         self.hift.load_state_dict(hift_state_dict, strict=True)
         self.hift.eval()