Fix no embed text spk tokens (#1540)
Signed-off-by: Junhong Liu <98734602+LJH-LBJ@users.noreply.github.com>
This commit is contained in:
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]
|
||||
|
||||
|
||||
Reference in new issue
Block a user