mirror of https://github.com/hpcaitech/ColossalAI
20 lines
408 B
Python
20 lines
408 B
Python
from .albert import *
|
|
from .bert import *
|
|
from .blip2 import *
|
|
from .bloom import *
|
|
from .chatglm2 import *
|
|
from .falcon import *
|
|
from .gpt import *
|
|
from .gptj import *
|
|
from .llama import *
|
|
from .opt import *
|
|
from .sam import *
|
|
from .t5 import *
|
|
from .vit import *
|
|
from .whisper import *
|
|
|
|
try:
|
|
from .mistral import *
|
|
except ImportError:
|
|
print("This version of transformers doesn't support mistral.")
|