diff --git a/tests/kit/model_zoo/transformers/__init__.py b/tests/kit/model_zoo/transformers/__init__.py index 029968231..4adc38619 100644 --- a/tests/kit/model_zoo/transformers/__init__.py +++ b/tests/kit/model_zoo/transformers/__init__.py @@ -2,8 +2,7 @@ from .albert import * from .bert import * from .blip2 import * from .bloom import * - -# from .chatglm2 import * +from .chatglm2 import * from .command import * from .deepseek import * from .falcon import *