github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/kit/model_zoo/transformers/__init__.py

8 lines
140 B
Python
Raw Normal View History

[test] added transformers models to test model zoo (#3135)
2023-03-15 03:26:10 +00:00
from .albert import *
from .bert import *
[shardformer] supported bloom model (#4098)
2023-06-28 07:04:35 +00:00
from .bloom import *
[test] added transformers models to test model zoo (#3135)
2023-03-15 03:26:10 +00:00
from .gpt import *
[shardformer] adapted T5 and LLaMa test to use kit (#4049) * [shardformer] adapted T5 and LLaMa test to use kit * polish code
2023-06-21 01:32:46 +00:00
from .llama import *
[test] added transformers models to test model zoo (#3135)
2023-03-15 03:26:10 +00:00
from .opt import *
from .t5 import *