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 | ||
| auto_policy.py | ||
| base_policy.py | ||
| bert.py | ||
| blip2.py | ||
| bloom.py | ||
| chatglm2.py | ||
| gpt2.py | ||
| llama.py | ||
| opt.py | ||
| sam.py | ||
| t5.py | ||
| vit.py | ||
| whisper.py | ||