Skip to content

vllm_omni.diffusion.models.soulx_singer.preprocess.pre_process

Shared pre_process_func helpers for SoulX-Singer pipelines (batch / full-payload only).

logger module-attribute

logger = init_logger(__name__)

attach_preprocess_for_diffusion_request

attach_preprocess_for_diffusion_request(
    request: OmniDiffusionRequest,
    *,
    kind: str,
    metadata_processor=None,
    sample_rate: int | None = None,
    device=None,
) -> OmniDiffusionRequest

Resolve preprocess payload for stage-1 diffusion (warmup / IPC / precomputed paths).

build_metadata_processor

build_metadata_processor(od_config: OmniDiffusionConfig)

build_precomputed_payload

build_precomputed_payload(
    kind: str,
    extra_args: dict[str, Any],
    *,
    metadata_processor,
    sample_rate: int | None,
    device,
) -> dict[str, Any]

build_preprocess_payload

build_preprocess_payload(
    kind: str,
    *,
    prompt: dict[str, Any],
    extra_args: dict[str, Any],
    preprocess: Module,
    metadata_processor,
    device,
    sample_rate: int,
) -> dict[str, Any]

build_warmup_payload

build_warmup_payload(
    kind: str,
    *,
    metadata_processor,
    device,
    sample_rate: int | None,
) -> dict[str, Any]

is_warmup_request

is_warmup_request(request: OmniDiffusionRequest) -> bool

normalize_svs_control_extra_args

normalize_svs_control_extra_args(
    extra_args: dict[str, Any],
) -> dict[str, Any]

resolve_preprocess_audio

resolve_preprocess_audio(
    prompt: dict[str, Any], extra_args: dict[str, Any]
) -> tuple[Any, Any]