[Refactor] Migrate Step-Audio2 online test to deploy config (#5309)
Signed-off-by: zwhzzz0821 <2831474076@qq.com>
This commit is contained in:
1 parent
8fc9df4331
commit
633ebbd8a4
4 files changed
+74
-113
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user