Zhongkai Zhao 70885d707d [hotfix] Suport extra_kwargs in ShardConfig (#5031) ... * [refactor]: replace inference args with extra_kwargs in ShardConfig * modify shardconfig * polish code * fix policy bug in llama * fix bug in auto policy * remove setattr in ShardConfig		2023-11-10 10:49:50 +08:00
..
__init__.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
shard_config.py	[hotfix] Suport extra_kwargs in ShardConfig (#5031)	2023-11-10 10:49:50 +08:00
sharder.py	[Inference] Fix bug in ChatGLM2 Tensor Parallelism (#5014)	2023-11-07 15:01:50 +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