42 lines
844 B
Python
42 lines
844 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,
|
|
PipelineConfig,
|
|
StageConfig,
|
|
StageDeployConfig,
|
|
StageExecutionType,
|
|
StagePipelineConfig,
|
|
StageType,
|
|
load_deploy_config,
|
|
merge_pipeline_deploy,
|
|
)
|
|
from vllm_omni.config.yaml_util import (
|
|
create_config,
|
|
load_yaml_config,
|
|
merge_configs,
|
|
to_dict,
|
|
)
|
|
|
|
__all__ = [
|
|
"OmniModelConfig",
|
|
"LoRAConfig",
|
|
"StageConfig",
|
|
"StageType",
|
|
"StageExecutionType",
|
|
"StagePipelineConfig",
|
|
"PipelineConfig",
|
|
"StageDeployConfig",
|
|
"DeployConfig",
|
|
"load_deploy_config",
|
|
"merge_pipeline_deploy",
|
|
"create_config",
|
|
"load_yaml_config",
|
|
"merge_configs",
|
|
"to_dict",
|
|
]
|