mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-22 17:58:56 +00:00
c6295c3381
* remove useless code * fix quant model * fix test import bug * mv original inference legacy * fix chatglm2
6 lines
225 B
Python
6 lines
225 B
Python
from .bloom import BloomInferenceForwards
|
|
from .chatglm2 import ChatGLM2InferenceForwards
|
|
from .llama import LlamaInferenceForwards
|
|
|
|
__all__ = ["LlamaInferenceForwards", "BloomInferenceForwards", "ChatGLM2InferenceForwards"]
|