a3903810fe
Signed-off-by: lishunyang <lishunyang12@163.com> Signed-off-by: reidliu41 <reid201711@gmail.com> Signed-off-by: Alex Brooks <albrooks@redhat.com> Co-authored-by: reidliu41 <reid201711@gmail.com> Co-authored-by: xiaohajiayou <75477391+xiaohajiayou@users.noreply.github.com> Co-authored-by: Alex Brooks <albrooks@redhat.com> Co-authored-by: Hongsheng Liu <liuhongsheng4@huawei.com>
48 lines
982 B
Python
48 lines
982 B
Python
"""
|
|
Configuration module for vLLM-Omni.
|
|
"""
|
|
|
|
from vllm_omni.config.lora import LoRAConfig
|
|
from vllm_omni.config.model import OmniModelConfig
|
|
from vllm_omni.config.stage_config import (
|
|
DeployConfig,
|
|
ModelPipeline,
|
|
PipelineConfig,
|
|
StageConfig,
|
|
StageConfigFactory,
|
|
StageDeployConfig,
|
|
StageExecutionType,
|
|
StagePipelineConfig,
|
|
StageType,
|
|
load_deploy_config,
|
|
merge_pipeline_deploy,
|
|
register_pipeline,
|
|
)
|
|
from vllm_omni.config.yaml_util import (
|
|
create_config,
|
|
load_yaml_config,
|
|
merge_configs,
|
|
to_dict,
|
|
)
|
|
|
|
__all__ = [
|
|
"OmniModelConfig",
|
|
"LoRAConfig",
|
|
"StageConfig",
|
|
"StageConfigFactory",
|
|
"ModelPipeline",
|
|
"StageType",
|
|
"StageExecutionType",
|
|
"StagePipelineConfig",
|
|
"PipelineConfig",
|
|
"StageDeployConfig",
|
|
"DeployConfig",
|
|
"load_deploy_config",
|
|
"merge_pipeline_deploy",
|
|
"register_pipeline",
|
|
"create_config",
|
|
"load_yaml_config",
|
|
"merge_configs",
|
|
"to_dict",
|
|
]
|