mirror of https://github.com/hpcaitech/ColossalAI
d367b88785
* [shardformer] fix opt test hanging * fix * test * test * test * fix test * fix test * remove print * add fix |
||
---|---|---|
.. | ||
__init__.py | ||
_utils.py | ||
test_shard_bert.py | ||
test_shard_blip2.py | ||
test_shard_bloom.py | ||
test_shard_chatglm2.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 |