flybird11111 0a25e16e46 [shardformer]gather llama logits (#5398) ... * gather llama logits * fix		2024-02-27 22:44:07 +08:00
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
shard_config.py	[shardformer]gather llama logits (#5398)	2024-02-27 22:44:07 +08:00
sharder.py	[nfc] fix typo colossalai/shardformer/ (#5133)	2024-01-04 16:21:55 +08:00
shardformer.py	[shardformer] rename policy file name	2023-08-15 23:25:14 +08:00
utils.py	[pipeline] update shardformer policy	2023-08-15 23:25:14 +08:00