Lekr0's picture
Add files using upload-large-folder tool
62dca4c verified
# 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",
]