File size: 367 Bytes
62dca4c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | 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",
]
|