Files
vllm-omni/vllm_omni/distributed/__init__.py
T
2026-09-13 16:35:02 +08:00

38 lines
944 B
Python

# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from .omni_connectors import (
ConnectorSpec,
MooncakeConnector,
MooncakeStoreConnector,
MooncakeTransferEngineConnector,
MoriTransferEngineConnector,
NixlConnector,
OmniConnectorBase,
OmniConnectorFactory,
OmniTransferConfig,
SharedMemoryConnector,
YuanrongConnector,
YuanrongTransferEngineConnector,
load_omni_transfer_config,
)
__all__ = [
# Config
"ConnectorSpec",
"OmniTransferConfig",
# Connectors
"OmniConnectorBase",
"OmniConnectorFactory",
"MooncakeConnector", # compat alias
"MooncakeStoreConnector",
"MooncakeTransferEngineConnector",
"MoriTransferEngineConnector",
"NixlConnector",
"SharedMemoryConnector",
"YuanrongConnector",
"YuanrongTransferEngineConnector",
# Utilities
"load_omni_transfer_config",
]