| from .base import Eagle3DraftModel | |
| from .dflash import ( | |
| DFlashDraftModel, | |
| build_target_layer_ids, | |
| extract_context_feature, | |
| sample, | |
| ) | |
| from .llama3_eagle import LlamaForCausalLMEagle3 | |
| __all__ = [ | |
| "Eagle3DraftModel", | |
| "DFlashDraftModel", | |
| "LlamaForCausalLMEagle3", | |
| "build_target_layer_ids", | |
| "extract_context_feature", | |
| "sample", | |
| ] | |