ColossalAI/tests/kit/model_zoo/transformers/__init__.py
Frank Lee 58df720570 [shardformer] adapted T5 and LLaMa test to use kit (#4049)
* [shardformer] adapted T5 and LLaMa test to use kit

* polish code
2023-07-04 16:05:01 +08:00

7 lines
119 B
Python

from .albert import *
from .bert import *
from .gpt import *
from .llama import *
from .opt import *
from .t5 import *