ColossalAI/tests/test_shardformer/test_model
FoolPlayer dfca9678fa integrate with dist layer (#4011) 2023-07-04 16:05:01 +08:00
..
test_shard_bert.py integrate with dist layer (#4011) 2023-07-04 16:05:01 +08:00
test_shard_llama.py [shardformer] support llama model using shardformer (#3969) 2023-07-04 16:05:01 +08:00
test_shard_t5.py [shardformer] shardformer support t5 model (#3994) 2023-07-04 16:05:01 +08:00