| from speculators.models.attention import ALL_ATTENTION_FUNCTIONS |
|
|
| from .dflash import DFlashDraftModel, DFlashSpeculatorConfig |
| from .dflash2 import DFlash2DraftModel, DFlash2SpeculatorConfig |
| from .dfly import DFlyDraftModel, DFlySpeculatorConfig |
| from .domino import DominoDraftModel, DominoSpeculatorConfig |
| from .dspark import DSparkDraftModel, DSparkSpeculatorConfig |
| from .eagle3 import Eagle3DraftModel, Eagle3SpeculatorConfig |
| from .mtp import MTPDraftModel, MTPSpeculatorConfig |
| from .ngram_candidates import NGramCandidatesDraftModel, NGramCandidatesSpeculatorConfig |
| from .ngram_dflash import NGramDFlashDraftModel, NGramDFlashSpeculatorConfig |
| from .peagle import PEagleDraftModel, PEagleSpeculatorConfig |
|
|
| __all__ = [ |
| "DFlash2DraftModel", |
| "DFlash2SpeculatorConfig", |
| "DFlashDraftModel", |
| "DFlashSpeculatorConfig", |
| "DFlyDraftModel", |
| "DFlySpeculatorConfig", |
| "DSparkDraftModel", |
| "DSparkSpeculatorConfig", |
| "DominoDraftModel", |
| "DominoSpeculatorConfig", |
| "Eagle3DraftModel", |
| "Eagle3SpeculatorConfig", |
| "MTPDraftModel", |
| "MTPSpeculatorConfig", |
| "NGramCandidatesDraftModel", |
| "NGramCandidatesSpeculatorConfig", |
| "NGramDFlashDraftModel", |
| "NGramDFlashSpeculatorConfig", |
| "PEagleDraftModel", |
| "PEagleSpeculatorConfig", |
| ] |
|
|
