Feature/chatglm (#4240)

* [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
This commit is contained in:
Kun Lin
2023-07-20 17:28:00 +08:00
committed by Hongxin Liu
parent 9ee4ebea83
commit ed34bb1310
6 changed files with 1672 additions and 0 deletions

View File

@@ -1,6 +1,7 @@
from .albert import *
from .bert import *
from .bloom import *
from .chatglm import *
from .gpt import *
from .llama import *
from .opt import *