汪志鹏
ccdb2b51aa
Signed-off-by: princepride <wangzhipeng628@gmail.com> Signed-off-by: 汪志鹏 <wangzhipeng628@gmail.com>
27 lines
775 B
Python
27 lines
775 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_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_x_to_text_model_family,
|
|
should_init_extra_args_for_non_diffusion_stages,
|
|
)
|
|
|
|
__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_x_to_text_model_family",
|
|
"should_init_extra_args_for_non_diffusion_stages",
|
|
]
|