Fix no embed text spk tokens (#1540)

Signed-off-by: Junhong Liu <98734602+LJH-LBJ@users.noreply.github.com>
This commit is contained in:
Junhong Liu authored and GitHub committed 2026-02-28 09:33:58 +08:00
1 parent 11f59f600d
commit 42d86af1f1
1 file changed
+1 -1
@@ -628,7 +628,7 @@ class Qwen2_5OmniForConditionalGeneration(
return set(["thinker_embedding.weight", "talker_embedding.weight"])
def _get_embed_text_spk_token(self, voice_type: str):
if voice_type not in self.embed_text_spk_tokens:
if not hasattr(self, "embed_text_spk_tokens") or voice_type not in self.embed_text_spk_tokens:
return self.embed_text_bos_token
return self.embed_text_spk_tokens[voice_type]