mirror of https://github.com/hpcaitech/ColossalAI
37d22f6878
* bloom policy * llama pipeline forward and tests * fix the output and attention_mask * fix name * bind argument to policy * finish bloom model * test shard gpt2 * clear cache |
||
---|---|---|
.. | ||
test_layer | ||
test_model | ||
__init__.py | ||
test_shard_utils.py | ||
test_with_torch_ddp.py |