..
__init__.py
[shardformer] adapted T5 and LLaMa test to use kit ( #4049 )
2023-07-04 16:05:01 +08:00
_utils.py
[shardformer] fix chatglm implementation ( #5644 )
2024-04-25 14:41:17 +08:00
test_shard_bert.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_blip2.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_bloom.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_chatglm2.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_falcon.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_gpt2.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_gptj.py
[shardformer] update colo attention to support custom mask ( #5510 )
2024-03-27 11:19:32 +08:00
test_shard_llama.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_mistral.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_opt.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_sam.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_t5.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_vit.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00
test_shard_whisper.py
[misc] refactor launch API and tensor constructor ( #5666 )
2024-04-29 10:40:11 +08:00