diff --git a/data/test/videos/action_recognition_test_video.mp4 b/data/test/videos/action_recognition_test_video.mp4 index d9f365b7..9197b770 100644 Binary files a/data/test/videos/action_recognition_test_video.mp4 and b/data/test/videos/action_recognition_test_video.mp4 differ diff --git a/modelscope/metainfo.py b/modelscope/metainfo.py index be6497f0..9adf2206 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):