ColossalAI/tests/kit
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
..
model_zoo [shardformer] support Blip2 (#4243) 2023-08-15 23:25:14 +08:00
__init__.py [test] added timm models to test model zoo (#3129) 2023-03-14 14:29:18 +08:00