File size: 584 Bytes
62dca4c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # from .auto import AutoDistributedTargetModel, AutoDraftModelConfig, AutoEagle3DraftModel
from .auto import AutoDraftModelConfig, AutoEagle3DraftModel
from .draft.llama3_eagle import LlamaForCausalLMEagle3
from .target.eagle3_target_model import (
CustomEagle3TargetModel,
HFEagle3TargetModel,
SGLangEagle3TargetModel,
get_eagle3_target_model,
)
__all__ = [
"LlamaForCausalLMEagle3",
"SGLangEagle3TargetModel",
"HFEagle3TargetModel",
"CustomEagle3TargetModel",
"get_eagle3_target_model",
"AutoDraftModelConfig",
"AutoEagle3DraftModel",
]
|