ColossalAI/colossalai/inference/tensor_parallel/modeling/__init__.py

5 lines
148 B
Python

from .bloom import BloomInferenceForwards
from .llama import LlamaInferenceForwards
__all__ = ['BloomInferenceForwards', 'LlamaInferenceForwards']