Skip to content

Commit 7b4ce83

Browse files
hjh0119Jintao-Huang
authored andcommitted
[vllm] support ovis vllm (#5566)
1 parent 38f2c15 commit 7b4ce83

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

swift/llm/template/template/qwen.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -745,11 +745,17 @@ class Ovis2_5Template(ThinkingTemplate):
745745
def replace_tag(self, media_type: Literal['image', 'video', 'audio'], index: int,
746746
inputs: StdTemplateInputs) -> List[Context]:
747747
if media_type == 'image':
748-
return [[-200], '\n']
748+
if self.mode == 'vllm':
749+
return ['<image>']
750+
else:
751+
return [[-200], '\n']
749752
elif media_type == 'video':
750-
num_frames = get_env_args('num_frames', int, self.num_frames)
751-
inputs.images = load_video_ovis2_5(inputs.videos[index], num_frames)
752-
return [[-200], '\n']
753+
if self.mode == 'vllm':
754+
return ['<video>']
755+
else:
756+
num_frames = get_env_args('num_frames', int, self.num_frames)
757+
inputs.images = load_video_ovis2_5(inputs.videos[index], num_frames)
758+
return [[-200], '\n']
753759

754760
def _encode(self, inputs: StdTemplateInputs) -> Dict[str, Any]:
755761
min_pixels = get_env_args('min_pixels', int, 448 * 448)

0 commit comments

Comments
 (0)