feat: support external custom models (#13429)

Co-authored-by: qiuxuan.lzw <qiuxuan.lzw@alibaba-inc.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
StonyPort
2025-11-20 11:16:58 +08:00
committed by GitHub
co-authored by qiuxuan.lzw Mick
parent 127d59cd2c
commit 48ca9f7518
8 changed files with 72 additions and 1 deletions
@@ -0,0 +1,21 @@
from sglang.srt.models.qwen2_vl import (
Qwen2VLForConditionalGeneration as OriginalQwen2VLForConditionalGeneration,
)
from sglang.srt.multimodal.processors.qwen_vl import QwenVLImageProcessor
class Qwen2VLForConditionalGeneration(OriginalQwen2VLForConditionalGeneration):
def __init__(self, config, quant_config, prefix: str = "") -> None:
super().__init__(config, quant_config, prefix)
print("init custom model:", self.__class__.__name__)
class CustomProcessor(QwenVLImageProcessor):
models = [Qwen2VLForConditionalGeneration]
def __init__(self, hf_config, server_args, _processor, *args, **kwargs):
super().__init__(hf_config, server_args, _processor, *args, **kwargs)
print("init custom processor:", self.__class__.__name__)
EntryClass = Qwen2VLForConditionalGeneration