From 42d86af1f1bd8e555f86c52be1738d273eb08951 Mon Sep 17 00:00:00 2001 From: Junhong Liu Date: Sat, 28 Feb 2026 09:33:58 +0800 Subject: [PATCH] Fix no embed text spk tokens (#1540) Signed-off-by: Junhong Liu <98734602+LJH-LBJ@users.noreply.github.com> --- vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni.py b/vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni.py index 39baa5573..726cc2643 100644 --- a/vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni.py +++ b/vllm_omni/model_executor/models/qwen2_5_omni/qwen2_5_omni.py @@ -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]