Allow Ace Step captioner to work with different output formats like yue2

This commit is contained in:
Jaret Burkett committed 2026-09-15 12:48:15 -06:00
1 parent 8bf12e476e
commit 6b2526a4d7
5 files changed
+75 -207

No files matched your search

+8
View File
@@ -120,8 +120,16 @@ class Qwen3OmniHandler(ArchHandler):
return key[len("thinker.") :]
class Qwen25OmniHandler(Qwen3OmniHandler):
"""Qwen2.5-Omni: same layout as Qwen3-Omni (thinker = text model + visual +
audio_tower + lm_head; talker / token2wav dropped). The thinker also
carries the speaker dict; only thinker.* tensors are kept."""
ARCH_HANDLERS = {
"Qwen3OmniMoeForConditionalGeneration": Qwen3OmniHandler,
"Qwen2_5OmniForConditionalGeneration": Qwen25OmniHandler,
"Qwen2_5OmniModel": Qwen25OmniHandler,
}