github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 12:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] added transformers models to test model zoo (#3135)

Browse Source
This commit is contained in:
Frank Lee
2023-03-15 11:26:10 +08:00
committed by GitHub
parent a674c63348
commit 6d48eb0560
12 changed files with 339 additions and 193 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/kit/model_zoo/transformers/__init__.py Normal file
View File

@@ -0,0 +1,5 @@
from .albert import *
from .bert import *
from .gpt import *
from .opt import *
from .t5 import *