32 lines
598 B
Python
32 lines
598 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 (
|
|
ModelPipeline,
|
|
StageConfig,
|
|
StageConfigFactory,
|
|
StageType,
|
|
)
|
|
from vllm_omni.config.yaml_util import (
|
|
create_config,
|
|
load_yaml_config,
|
|
merge_configs,
|
|
to_dict,
|
|
)
|
|
|
|
__all__ = [
|
|
"OmniModelConfig",
|
|
"LoRAConfig",
|
|
"StageConfig",
|
|
"StageConfigFactory",
|
|
"ModelPipeline",
|
|
"StageType",
|
|
"create_config",
|
|
"load_yaml_config",
|
|
"merge_configs",
|
|
"to_dict",
|
|
]
|