vllm-omni/vllm_omni/config/__init__.py

"""
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",
]