mirror of https://github.com/hpcaitech/ColossalAI
8739aa7fa0
* add some base tests and policies * finish whisper base model * add conditional generation * finish basic tests * whisper * finish whisper * finish whisper * del useless whisper test * fix * add argmin to replace * finish revision |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _utils.py | ||
| test_shard_bert.py | ||
| test_shard_blip2.py | ||
| test_shard_bloom.py | ||
| test_shard_chatglm.py | ||
| test_shard_gpt2.py | ||
| test_shard_llama.py | ||
| test_shard_opt.py | ||
| test_shard_sam.py | ||
| test_shard_t5.py | ||
| test_shard_vit.py | ||
| test_shard_whisper.py | ||