mirror of https://github.com/hpcaitech/ColossalAI
70885d707d
* [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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| llama.py | ||