mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-07-12 19:06:17 +00:00
ce7ade3882
* add chatglm2 * add * gather needed kernels * fix some bugs * finish context forward * finish context stage * fix * add * pause * add * fix bugs * finish chatglm * fix bug * change some logic * fix bugs * change some logics * add * add * add * fix * fix tests * fix
8 lines
240 B
Python
8 lines
240 B
Python
import _utils
|
|
|
|
from .bloom import BloomInferenceForwards
|
|
from .chatglm2 import ChatGLM2InferenceForwards
|
|
from .llama import LlamaInferenceForwards
|
|
|
|
__all__ = ["BloomInferenceForwards", "LlamaInferenceForwards", "ChatGLM2InferenceForwards"]
|