mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-14 05:33:23 +00:00
[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
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from .albert import *
|
||||
from .bert import *
|
||||
from .blip2 import *
|
||||
from .bloom import *
|
||||
from .chatglm import *
|
||||
from .gpt import *
|
||||
|
Reference in New Issue
Block a user