ColossalAI/colossalai/inference/pipeline/policies/__init__.py

4 lines
78 B
Python

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