Taichang ZhouandClaude Opus 4.8
9678ae58cf
Squashed migration rebased onto current main. Collapses the 46-commit dev/migrate-MR-v2 history (10 merge commits + superseded pre-rebase work) into a single commit, so the diff reflects only the genuine MR-V2 feature (worker_v2/ runners + model_states, omni schedulers, stage engine pipeline, pooling_output serialization bridge). Content already merged to main separately (step_audio2, tts_adapters, npu/_310p, etc.) drops out. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
18 lines
685 B
Python
18 lines
685 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import inspect
|
|
|
|
from vllm_omni.worker_v2.omni_ar_model_runner import OmniARModelRunner
|
|
from vllm_omni.worker_v2.omni_generation_model_runner import OmniGenerationModelRunner
|
|
from vllm_omni.worker_v2.omni_model_runner import OmniGPUModelRunner
|
|
|
|
|
|
def test_v2_model_runners_accept_vllm_020_is_profile_keyword():
|
|
for runner_cls in (OmniARModelRunner, OmniGenerationModelRunner, OmniGPUModelRunner):
|
|
assert "is_profile" in inspect.signature(runner_cls.execute_model).parameters
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_v2_model_runners_accept_vllm_020_is_profile_keyword()
|