Sy03andYueqian Lin
58007c0084
Signed-off-by: Sysy <1370724210@qq.com> Co-authored-by: Yueqian Lin <70319226+linyueqian@users.noreply.github.com>
15 lines
521 B
Python
15 lines
521 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from pathlib import Path
|
|
|
|
import yaml
|
|
|
|
|
|
def test_indextts25_default_recipe_enables_validated_cfm_batching():
|
|
recipe_path = Path(__file__).parents[4] / "vllm_omni" / "deploy" / "indextts2_5.yaml"
|
|
recipe = yaml.safe_load(recipe_path.read_text(encoding="utf-8"))
|
|
stage1 = next(stage for stage in recipe["stages"] if stage["stage_id"] == 1)
|
|
|
|
assert stage1["hf_overrides"]["s2mel_cfm_batch_size"] == 4
|