rein yang
afbcc51672
Signed-off-by: Rein Yang <ruiruyang2@gmail.com> Signed-off-by: R2-Y <ruiruyang2@gmail.com>
70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from vllm_omni.model_executor.models.aura_omni.qwen3_vl import (
|
|
AuraQwen3VLForConditionalGeneration,
|
|
AuraQwen3VLProcessingInfo,
|
|
Qwen3VLProcessor,
|
|
)
|
|
from vllm_omni.model_executor.models.registry import OmniModelRegistry
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
|
|
|
|
|
class _FakeCtx:
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self.processor_type = None
|
|
|
|
def get_hf_config(self, typ=None):
|
|
if typ is not None:
|
|
raise TypeError("wrong config class")
|
|
return self.config
|
|
|
|
def get_hf_processor(self, typ=None, **kwargs):
|
|
self.processor_type = typ
|
|
return SimpleNamespace(kwargs=kwargs)
|
|
|
|
|
|
def test_aura_qwen3_vl_processing_accepts_remote_qwen3_vl_config():
|
|
info = object.__new__(AuraQwen3VLProcessingInfo)
|
|
info.ctx = _FakeCtx(
|
|
SimpleNamespace(
|
|
model_type="qwen3_vl",
|
|
vision_config=SimpleNamespace(spatial_merge_size=2),
|
|
text_config=SimpleNamespace(hidden_size=4096),
|
|
)
|
|
)
|
|
|
|
assert info.get_hf_config().model_type == "qwen3_vl"
|
|
|
|
|
|
def test_aura_qwen3_vl_processing_rejects_unrelated_remote_config():
|
|
info = object.__new__(AuraQwen3VLProcessingInfo)
|
|
info.ctx = _FakeCtx(SimpleNamespace(model_type="not_qwen3_vl"))
|
|
|
|
with pytest.raises(TypeError, match="wrong config class"):
|
|
info.get_hf_config()
|
|
|
|
|
|
def test_aura_qwen3_vl_model_arch_registered():
|
|
model_cls = OmniModelRegistry._try_load_model_cls("AuraQwen3VLForConditionalGeneration")
|
|
|
|
assert model_cls is AuraQwen3VLForConditionalGeneration
|
|
|
|
|
|
def test_aura_qwen3_vl_processing_forces_upstream_processor_class():
|
|
info = object.__new__(AuraQwen3VLProcessingInfo)
|
|
ctx = _FakeCtx(SimpleNamespace(model_type="qwen3_vl"))
|
|
info.ctx = ctx
|
|
|
|
processor = info.get_hf_processor(extra="value")
|
|
|
|
assert ctx.processor_type is Qwen3VLProcessor
|
|
assert processor.kwargs["use_fast"] is True
|
|
assert processor.kwargs["extra"] == "value"
|