ColossalAI/tests/kit
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
..
model_zoo [shardformer] fix chatglm implementation (#5644) 2024-04-25 14:41:17 +08:00
__init__.py [test] added timm models to test model zoo (#3129) 2023-03-14 14:29:18 +08:00