mirror of https://github.com/hpcaitech/ColossalAI
083d7da33d
* complete policy for T5Model & T5ForConditionalGeneration * modify function signature in forwards * add forward for T5model * add forward for T5ForConditionalGeneration * fix a bug * fix hidden_states transporting in decoder * fix the passing of encoder_outputs |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _utils.py | ||
| test_pure_pipeline.py | ||
| test_shard_bert.py | ||
| test_shard_bert_pipeline.py | ||
| test_shard_bloom.py | ||
| test_shard_bloom_pipeline.py | ||
| test_shard_gpt2.py | ||
| test_shard_gpt2_pipeline.py | ||
| test_shard_llama.py | ||
| test_shard_llama_pipeline.py | ||
| test_shard_opt.py | ||
| test_shard_opt_pipeline.py | ||
| test_shard_t5.py | ||
| test_shard_t5_pipeline.py | ||
| test_shard_vit.py | ||