ColossalAI/tests/kit/model_zoo/transformers/__init__.py

13 lines
246 B
Python

from .albert import *
from .bert import *
from .blip2 import *
from .bloom import *
from .chatglm2 import *
from .gpt import *
from .llama import *
from .opt import *
from .sam import *
from .t5 import *
from .vit import *
from .whisper import *