mirror of https://github.com/hpcaitech/ColossalAI
31 lines
647 B
Python
31 lines
647 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.")
|
|
|
|
try:
|
|
from .qwen2 import *
|
|
except ImportError:
|
|
print("This version of transformers doesn't support qwen2.")
|
|
|
|
|
|
try:
|
|
from .command import *
|
|
except ImportError:
|
|
print("This version of transformers doesn't support Command-R.")
|