20 lines
583 B
Python
20 lines
583 B
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||
|
|
|
||
|
|
from vllm_omni.model_extras.registry import (
|
||
|
|
build_image_to_image_prompt,
|
||
|
|
build_text_to_image_prompt,
|
||
|
|
get_extra_body_params,
|
||
|
|
get_extra_output_params,
|
||
|
|
get_model_class_name,
|
||
|
|
should_init_extra_args_for_non_diffusion_stages,
|
||
|
|
)
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
"build_image_to_image_prompt",
|
||
|
|
"build_text_to_image_prompt",
|
||
|
|
"get_extra_body_params",
|
||
|
|
"get_extra_output_params",
|
||
|
|
"get_model_class_name",
|
||
|
|
"should_init_extra_args_for_non_diffusion_stages",
|
||
|
|
]
|