diff --git a/modelscope/metainfo.py b/modelscope/metainfo.py index 40f57f1a..cea10739 100644 --- a/modelscope/metainfo.py +++ b/modelscope/metainfo.py @@ -13,7 +13,7 @@ class Models(object): # nlp models bert = 'bert' - palm2_0 = 'palm2.0' + palm = 'palm-v2' structbert = 'structbert' # audio models diff --git a/modelscope/models/nlp/palm_for_text_generation.py b/modelscope/models/nlp/palm_for_text_generation.py index f4518d4f..9acd005b 100644 --- a/modelscope/models/nlp/palm_for_text_generation.py +++ b/modelscope/models/nlp/palm_for_text_generation.py @@ -8,7 +8,7 @@ from ..builder import MODELS __all__ = ['PalmForTextGeneration'] -@MODELS.register_module(Tasks.text_generation, module_name=Models.palm2_0) +@MODELS.register_module(Tasks.text_generation, module_name=Models.palm) class PalmForTextGeneration(Model): def __init__(self, model_dir: str, *args, **kwargs):