mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-11 22:54:10 +00:00
ed34bb1310
* [shardformer] added tests * [shardformer] vit test finish and support * [shardformer] chatglm ready * import chatglm * [shardformer] add test kit in model zoo for chatglm * [sharformer] add first version of policy of chatglm * [shardformer] polish chatglm code * [shardformer] polish code * [shardformer] support chatglm without layernorm * [shardformer] chatglm shard without mlp sharding * [shardformer] delete some file * [shardformer] ChatGLM support layernorm sharding * [shardformer] register without auto policy * [shardformer] pre-commit check files * [shardformer] fix chatglm configuration with pre-commit
12 lines
224 B
Python
12 lines
224 B
Python
from .albert import *
|
|
from .bert import *
|
|
from .bloom import *
|
|
from .chatglm import *
|
|
from .gpt import *
|
|
from .llama import *
|
|
from .opt import *
|
|
from .sam import *
|
|
from .t5 import *
|
|
from .vit import *
|
|
from .whisper import *
|