ColossalAI/tests/kit/model_zoo/transformers
flybird11111 1edc9b5fb3 [shardformer] update tests for all optimization (#4413)
[shardformer] update tests for all optimization
2023-08-15 23:25:14 +08:00
..
__init__.py [shardformer] support Blip2 (#4243) 2023-08-15 23:25:14 +08:00
albert.py [test] added transformers models to test model zoo (#3135) 2023-03-15 11:26:10 +08:00
bert.py [shardformer] update tests for all optimization (#4413) 2023-08-15 23:25:14 +08:00
blip2.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 2023-08-15 23:25:14 +08:00
bloom.py [shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395) 2023-08-15 23:25:14 +08:00
chatglm.py [shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395) 2023-08-15 23:25:14 +08:00
gpt.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 2023-08-15 23:25:14 +08:00
llama.py [shardformer] adapted T5 and LLaMa test to use kit (#4049) 2023-07-04 16:05:01 +08:00
opt.py [test] Hotfix/fix some model test and refactor check util api (#4369) 2023-08-15 23:25:14 +08:00
sam.py [shardformer] support SAM (#4231) 2023-08-15 23:25:14 +08:00
t5.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 2023-08-15 23:25:14 +08:00
vit.py [shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395) 2023-08-15 23:25:14 +08:00
whisper.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 2023-08-15 23:25:14 +08:00