wehub-resource-sync
26 lines
888 B
Python
26 lines
888 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# DeepSpeed Team
|
|
"""Compatibility shim for AutoEP preset adapter APIs."""
|
|
|
|
from deepspeed.module_inject.auto_ep_presets.base import (
|
|
AutoEPPresetAdapter,
|
|
ForwardContract,
|
|
GroupRoutingConfig,
|
|
TransformersTopLevelRouterLogitsAdapter,
|
|
)
|
|
from deepspeed.module_inject.auto_ep_presets.deepseek_v2 import DeepSeekV2PresetAdapter
|
|
from deepspeed.module_inject.auto_ep_presets.deepseek_v3 import DeepSeekV3PresetAdapter
|
|
from deepspeed.module_inject.auto_ep_presets.qwen3_5_moe import Qwen35MoePresetAdapter
|
|
from deepspeed.module_inject.auto_ep_presets.registry import get_preset_adapter
|
|
|
|
__all__ = [
|
|
"AutoEPPresetAdapter",
|
|
"DeepSeekV2PresetAdapter",
|
|
"DeepSeekV3PresetAdapter",
|
|
"ForwardContract",
|
|
"GroupRoutingConfig",
|
|
"Qwen35MoePresetAdapter",
|
|
"TransformersTopLevelRouterLogitsAdapter",
|
|
"get_preset_adapter",
|
|
]
|