ColossalAI/colossalai/inference/spec/__init__.py

5 lines
129 B
Python

from .drafter import Drafter
from .struct import DrafterOutput, GlideInput
__all__ = ["Drafter", "DrafterOutput", "GlideInput"]