update

modules/generate_audio.py

@@ -69,12 +69,14 @@ def generate_audio_batch(
     if isinstance(spk, int):
         with SeedContext(spk):
             params_infer_code["spk_emb"] = chat_tts.sample_random_speaker()
-        logger.info(("spk", spk))
+        logger.debug(("spk", spk))
     elif isinstance(spk, Speaker):
         params_infer_code["spk_emb"] = spk.emb
-        logger.info(("spk", spk.name))
+        logger.debug(("spk", spk.name))
+    else:
+        raise ValueError("spk must be int or Speaker")
 
-    logger.info(
+    logger.debug(
         {
             "text": texts,
             "infer_seed": infer_seed,

modules/speaker.py

@@ -106,8 +106,10 @@ class SpeakerManager:
         return speaker
 
     def get_speaker(self, name) -> Speaker | None:
-        filename = name + ".pt"
-        return self.speakers.get(filename, None)
+        for speaker in self.speakers.values():
+            if speaker.name == name:
+                return speaker
+        return None
 
     def get_speaker_by_id(self, id) -> Speaker | None:
         for speaker in self.speakers.values():