ColossalAI/tests/kit/model_zoo/transformers
FoolPlayer 879301d0da [shardformer] support Blip2 (#4243)
* support base blip2

* add support for downstream blip2 model

* update readme

* add forward injection

* skip not compatible models test

* fix test for gemini and low_level_zero_pugin
2023-08-15 23:25:14 +08:00
..
chatglm2_6b [shardformer] pre-commit check files 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 [pipeline] All bert models (#4233) 2023-08-15 23:25:14 +08:00
blip2.py [shardformer] support Blip2 (#4243) 2023-08-15 23:25:14 +08:00
bloom.py [shardformer] supported bloom model (#4098) 2023-07-04 16:05:01 +08:00
chatglm.py [shardformer] support ChatGLMForConditionalGeneration & add fusedlayernorm for vit 2023-08-15 23:25:14 +08:00
gpt.py [pipeline] support fp32 for HybridPlugin/merge shardformer test and pipeline test into one file (#4354) 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 [shardformer] shardformer support opt models (#4091) 2023-07-04 16:05:01 +08:00
sam.py [shardformer] support SAM (#4231) 2023-08-15 23:25:14 +08:00
t5.py [shardformer] adapted T5 and LLaMa test to use kit (#4049) 2023-07-04 16:05:01 +08:00
vit.py [shardformer] support pipeline base vit model (#4284) 2023-08-15 23:25:14 +08:00
whisper.py [shardformer] support whisper (#4212) 2023-08-15 23:25:14 +08:00