ColossalAI/colossalai/inference/__init__.py
2024-02-02 14:31:10 +08:00

5 lines
120 B
Python

from .config import InferenceConfig
from .core import InferenceEngine
__all__ = ["InferenceConfig", "InferenceEngine"]