ColossalAI/colossalai/legacy/inference/hybridengine/modeling/__init__.py

4 lines
80 B
Python

from .llama import LlamaInferenceForwards
__all__ = ["LlamaInferenceForwards"]