[fix] fix model zoo init

hpcaitech · Nov 1, 2024 · 8e40087 · 8e40087
1 parent 0218e67
commit 8e40087
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/kit/model_zoo/transformers/__init__.py b/tests/kit/model_zoo/transformers/__init__.py
@@ -2,8 +2,7 @@
 from .bert import *
 from .blip2 import *
 from .bloom import *
-
-# from .chatglm2 import *
+from .chatglm2 import *
 from .command import *
 from .deepseek import *
 from .falcon import *