Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
60165f24be |
No files matched your search
@@ -229,14 +229,6 @@ class TestDeployTopology:
|
||||
"vllm_omni.model_executor.stage_input_processors.minicpmo_4_5_omni.tts2code2wav_token_only"
|
||||
)
|
||||
|
||||
def test_no_async_chunk_selects_full_payload_processor(self) -> None:
|
||||
deploy = load_deploy_config(_DEPLOY_DIR / "minicpmo_4_5.yaml")
|
||||
deploy.async_chunk = False
|
||||
stages = merge_pipeline_deploy(OMNI_PIPELINES[_PIPELINE_KEY], deploy)
|
||||
|
||||
assert stages[1].yaml_engine_args["async_chunk"] is False
|
||||
assert stages[1].yaml_engine_args["custom_process_next_stage_input_func"].endswith(".tts2code2wav_full_payload")
|
||||
|
||||
|
||||
def test_code2wav_model_is_lazily_registered() -> None:
|
||||
assert _OMNI_MODELS["MiniCPMO45Code2Wav"] == (
|
||||
|
||||
@@ -72,16 +72,6 @@ def _codes(payload) -> list[int]:
|
||||
return payload.codes.audio.tolist()
|
||||
|
||||
|
||||
def test_empty_full_payload_releases_consumer_wait_gate() -> None:
|
||||
payload = tts2code2wav_full_payload(_manager(), None, _request("req"))
|
||||
|
||||
assert _codes(payload) == []
|
||||
assert payload.meta.code_flat_numel == 0
|
||||
assert payload.meta.left_context_size == 0
|
||||
assert payload.meta.last_chunk is True
|
||||
assert payload.meta.finished.item() is True
|
||||
|
||||
|
||||
@pytest.mark.parametrize(("count", "emitted"), [(24, False), (25, True), (26, True)])
|
||||
def test_first_chunk_threshold_is_25_generated_codes(count: int, emitted: bool) -> None:
|
||||
manager = _manager()
|
||||
|
||||
Reference in new issue
Block a user