mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-07-14 22:28:18 +00:00
6 lines
225 B
Python
6 lines
225 B
Python
from .bloom import BloomInferenceForwards
|
|
from .chatglm2 import ChatGLM2InferenceForwards
|
|
from .llama import LlamaInferenceForwards
|
|
|
|
__all__ = ["BloomInferenceForwards", "LlamaInferenceForwards", "ChatGLM2InferenceForwards"]
|