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

[fix] fix model zoo init

Browse Source
This commit is contained in:
duanjunwen 2024-11-01 09:02:07 +00:00
parent 0218e673db
commit 8e40087633
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/kit/model_zoo/transformers/__init__.py
View File

@ -2,8 +2,7 @@ from .albert import *
from .bert import * from .bert import *
from .blip2 import * from .blip2 import *
from .bloom import * from .bloom import *
from .chatglm2 import *
# from .chatglm2 import *
from .command import * from .command import *
from .deepseek import * from .deepseek import *
from .falcon import * from .falcon import *