a7ab91fb85
Signed-off-by: suyanli220 <suyanli220@gmail.com> Signed-off-by: suyan.li <suyan.li@bytedance.com> Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com> Co-authored-by: suyan.li <suyan.li@bytedance.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
39 lines
1.2 KiB
Python
39 lines
1.2 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_ar_input_builder,
|
|
get_ar_tokenizer_validator,
|
|
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_ar_input_builder",
|
|
"get_ar_tokenizer_validator",
|
|
"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",
|
|
]
|