14 lines
521 B
Python
14 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
|