ColossalAI/tests/kit/model_zoo
Hongxin Liu bbb2c21f16
[shardformer] fix chatglm implementation (#5644)
* [shardformer] fix chatglm policy

* [shardformer] fix chatglm flash attn

* [shardformer] update readme

* [shardformer] fix chatglm init

* [shardformer] fix chatglm test

* [pipeline] fix chatglm merge batch
2024-04-25 14:41:17 +08:00
..
custom [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
diffusers [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
timm [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchaudio [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
torchrec [test] modify model supporting part of low_level_zero plugin (including correspoding docs) 2023-10-05 15:10:31 +08:00
torchvision [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
transformers [shardformer] fix chatglm implementation (#5644) 2024-04-25 14:41:17 +08:00
__init__.py [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00
executor.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00
registry.py [devops] remove post commit ci (#5566) 2024-04-08 15:09:40 +08:00