Update cosyvoice/cli/frontend.py

cosyvoice/cli/frontend.py

@@ -112,7 +112,7 @@ class CosyVoiceFrontEnd:
     def frontend_sft(self, tts_text, spk_id):
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         tts_text_token, tts_text_token_len = self._extract_text_token(tts_text)
-        embedding = self.spk2info[spk_id]['embedding']
+        embedding = self.spk2info[spk_id]['embedding'].to(self.device)
         model_input = {'text': tts_text_token, 'text_len': tts_text_token_len, 'llm_embedding': embedding, 'flow_embedding': embedding}
         return model_input