mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-07-13 20:39:59 +00:00
b3f5d7a3ba
* Feature/vit support (#4182) * [shardformer] added tests * [shardformer] vit test finish and support * fix attention dropout * support base vit pipeline * support vit downstream model * fix vit shard test * modify hidden states return type --------- Co-authored-by: Kun Lin <81014421+klhhhhh@users.noreply.github.com>
9 lines
159 B
Python
9 lines
159 B
Python
from .albert import *
|
|
from .bert import *
|
|
from .bloom import *
|
|
from .gpt import *
|
|
from .llama import *
|
|
from .opt import *
|
|
from .t5 import *
|
|
from .vit import *
|