Skip to content

Commit 4dfbf44

Browse files
authored
[qwen2.5-omni] Fix omni get_template (#4518)
1 parent 274bc4a commit 4dfbf44

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

swift/llm/template/template/emu3.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,10 @@ class Emu3GenTemplate(Template):
2727
'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, '
2828
'worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry.')
2929

30-
def __init__(self, *args, **kwargs):
31-
super().__init__(*args, **kwargs)
30+
def init_processor(self, processor) -> None:
31+
if processor is None:
32+
return
33+
super().init_processor(processor)
3234
self.bov = self.processor.tokenizer.encode(self.processor.visual_template[0].format(token_id=0))[0]
3335
self.eov = self.processor.tokenizer.encode(self.processor.visual_template[0].format(token_id=self.COOKBOOK_SIZE
3436
- 1))[0]

swift/llm/template/template/qwen.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -408,8 +408,10 @@ class Qwen2_5OmniTemplate(Qwen2_5VLTemplate):
408408
version = 'omni'
409409
placeholder_tokens = ['<|IMAGE|>', '<|AUDIO|>', '<|VIDEO|>']
410410

411-
def __init__(self, *args, **kwargs):
412-
super().__init__(*args, **kwargs)
411+
def init_processor(self, processor) -> None:
412+
if processor is None:
413+
return
414+
super().init_processor(processor)
413415
from transformers.models.qwen2_5_omni.processing_qwen2_5_omni import Qwen2_5OmniProcessorKwargs
414416
default = Qwen2_5OmniProcessorKwargs._defaults
415417
self.seconds_per_chunk = default['videos_kwargs']['seconds_per_chunk']

0 commit comments

Comments
 (0)