未验证 提交 4382901c 编写于 作者: K Kennycao123 提交者: GitHub

Merge pull request #811 from Kennycao123/ernie-kit-open-v1.0

fix path
......@@ -64,8 +64,8 @@ class RegisterSet(object):
embedding = Register("embedding")
inference = Register("inference")
package_names = ['wenxin.data.field_reader', 'wenxin.data.data_set_reader', 'wenxin.model',
'wenxin.data.tokenizer', 'wenxin.modules.token_embedding']
package_names = ['erniekit.data.field_reader', 'erniekit.data.data_set_reader', 'erniekit.model',
'erniekit.data.tokenizer', 'erniekit.modules.token_embedding']
ALL_MODULES = []
for package_name in package_names:
......@@ -83,7 +83,7 @@ class RegisterSet(object):
def import_modules_plugin():
"""import ernie 插件版需要的包,结合注册机制用
"""
PLUGIN_MODULES = [("wenxin.data.tokenizer", "tokenization_wp"), ("wenxin.modules", "ernie_origin")]
PLUGIN_MODULES = [("erniekit.data.tokenizer", "tokenization_wp"), ("erniekit.modules", "ernie_origin")]
for package_name, file_name in PLUGIN_MODULES:
import_new_module(package_name, file_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册