Hongsheng Liu
d4096e01fe
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com> Signed-off-by: hsliuustc0106 <liuhongsheng4@huawei.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
|
|
from vllm_omni.model_extras.registry import (
|
|
build_image_to_image_prompt,
|
|
build_image_to_video_prompt,
|
|
build_text_to_image_prompt,
|
|
build_x_to_text_prompt,
|
|
get_extra_body_params,
|
|
get_extra_output_params,
|
|
get_model_class_name,
|
|
get_output_tensor_range,
|
|
get_transformer_config_subfolder,
|
|
get_video_generation_defaults,
|
|
get_x_to_text_model_family,
|
|
should_init_extra_args_for_non_diffusion_stages,
|
|
should_preserve_reference_image_size,
|
|
)
|
|
|
|
__all__ = [
|
|
"build_image_to_image_prompt",
|
|
"build_image_to_video_prompt",
|
|
"build_text_to_image_prompt",
|
|
"build_x_to_text_prompt",
|
|
"get_extra_body_params",
|
|
"get_extra_output_params",
|
|
"get_model_class_name",
|
|
"get_output_tensor_range",
|
|
"get_transformer_config_subfolder",
|
|
"get_video_generation_defaults",
|
|
"get_x_to_text_model_family",
|
|
"should_init_extra_args_for_non_diffusion_stages",
|
|
"should_preserve_reference_image_size",
|
|
]
|