ColossalAI/colossalai/pipeline/schedule
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
..
__init__.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
_utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
base.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
generate.py [hotfix] fix typo s/keywrods/keywords etc. (#5429) 2024-03-12 11:25:16 +08:00
interleaved_pp.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
one_f_one_b.py [shardformer] fix chatglm implementation (#5644) 2024-04-25 14:41:17 +08:00