github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[fix] fix model zoo import;

pull/6034/head
duanjunwen 2024-09-09 06:39:33 +00:00
parent a5ec3d4285
commit fed8b1587d
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 *