28 lines
882 B
Python
28 lines
882 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
from __future__ import annotations
|
|
|
|
# Request params the AudioX pipeline reads from ``sampling_params.extra_args``
|
|
# (see ``pipeline_audiox.py`` ``forward``). Declaring them here routes
|
|
# ``extra_body`` (online) / ``--extra-body`` (offline) keys into ``extra_args``
|
|
# instead of silently dropping them.
|
|
AUDIOX_EXTRA_BODY_PARAMS = frozenset(
|
|
{
|
|
"audiox_task", # t2a/t2m/v2a/v2m/tv2a/tv2m
|
|
"seconds_start",
|
|
"seconds_total",
|
|
"sigma_min",
|
|
"sigma_max",
|
|
"cfg_rescale",
|
|
"video_path", # v2*/tv2* conditioning
|
|
"audio_path", # optional reference-audio conditioning
|
|
}
|
|
)
|
|
# Echoed back in diffusion output metadata (pipeline_audiox.py).
|
|
AUDIOX_EXTRA_OUTPUT_PARAMS = frozenset(
|
|
{
|
|
"audiox_task",
|
|
}
|
|
)
|