[Refactor] Migrate Step-Audio2 online test to deploy config (#5309)

Signed-off-by: zwhzzz0821 <2831474076@qq.com>
This commit is contained in:
Yukim1 authored and GitHub committed 2026-07-22 09:44:27 +00:00
1 parent 8fc9df4331
commit 633ebbd8a4
4 files changed
+74 -113

No files matched your search

+28
View File
@@ -1174,6 +1174,7 @@ stages:
("step_audio_2.yaml", "step_audio_2", 2, "audio", None),
("step_audio_2_asr.yaml", "step_audio_2_asr", 1, "text", "step_audio_2_asr"),
("step_audio_2_async_chunk.yaml", "step_audio_2", 2, "audio", None),
("step_audio2_ci.yaml", "step_audio_2_asr", 1, "text", "step_audio_2_asr"),
("hunyuan_video_15.yaml", "hunyuan_video_15", 1, "video", None),
("wan2_2_ti2v.yaml", "wan2_2_ti2v", 1, "video", None),
],
@@ -1241,6 +1242,33 @@ stages:
assert deploy.stages[0].tensor_parallel_size == 2
assert deploy.stages[1].devices == "1"
def test_step_audio2_online_ci_deploy_is_single_thinker(self):
deploy = load_deploy_config(Path(get_deploy_config_path("step_audio2_ci.yaml")))
assert deploy.pipeline == "step_audio_2_asr"
assert deploy.async_chunk is False
assert deploy.trust_remote_code is True
assert deploy.enable_prefix_caching is False
assert len(deploy.stages) == 1
stage = deploy.stages[0]
assert stage.devices == "0"
assert stage.tensor_parallel_size == 1
assert stage.max_model_len == 1024
assert stage.max_num_batched_tokens == 1024
assert stage.max_num_seqs == 1
assert stage.gpu_memory_utilization == 0.7
assert stage.skip_mm_profiling is True
assert stage.enforce_eager is True
assert stage.async_scheduling is False
assert stage.engine_extras["hf_overrides"]["architectures"] == ["StepAudio2ThinkerForConditionalGeneration"]
stages = merge_pipeline_deploy(OMNI_PIPELINES[deploy.pipeline], deploy)
assert len(stages) == 1
assert stages[0].final_output is True
assert stages[0].final_output_type == "text"
assert stages[0].yaml_engine_args["engine_output_type"] == "text"
def test_step_audio2_dispatches_sync_and_async_chunk_processors(self):
pipeline = resolve_pipeline_config("step_audio_2")
assert isinstance(pipeline, PipelineConfig)