Skip to content

vllm_omni.diffusion.sched.base_scheduler

logger module-attribute

logger = init_logger(__name__)

get_request_batch_sampling_params_key

get_request_batch_sampling_params_key(
    request: OmniDiffusionRequest,
) -> RequestBatchSamplingParamsKey

Build a request-batch compatibility key from the request's sampling params.

get_sampling_params_key

get_sampling_params_key(
    request: OmniDiffusionRequest,
) -> SamplingParamsKey

Build a batch-compatibility key from the request's sampling params.