You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/kit/model_zoo/transformers
flybird11111 59e252ecdb
[shardformer] chatglm support sequence parallel (#4482)
1 year ago
..
__init__.py rename chatglm to chatglm2 (#4484) 1 year ago
albert.py
bert.py [shardformer] update tests for all optimization (#4413) 1 year ago
blip2.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 1 year ago
bloom.py [shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395) 1 year ago
chatglm2.py [shardformer] chatglm support sequence parallel (#4482) 1 year ago
gpt.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 1 year ago
llama.py [shardformer] adapted T5 and LLaMa test to use kit (#4049) 1 year ago
opt.py [test] Hotfix/fix some model test and refactor check util api (#4369) 1 year ago
sam.py [shardformer] support SAM (#4231) 1 year ago
t5.py [shardformer]update t5 tests for using all optimizations. (#4407) 1 year ago
vit.py [shardformer] rewrite tests for opt/bloom/llama/vit/chatglm (#4395) 1 year ago
whisper.py [Shardformer] Merge flash attention branch to pipeline branch (#4362) 1 year ago