Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-09-01 19:49:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] update nlp models name in metainfo

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9134657
This commit is contained in:
zhangzhicheng.zzc
2022-06-23 11:58:22 +08:00
committed by yingda.chen
parent ace8af9246
commit 371530ed1d
3 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
data/test/videos/action_recognition_test_video.mp4
View File

Binary file not shown.

2
modelscope/metainfo.py
View File

@@ -13,7 +13,7 @@ class Models(object):
# nlp models
bert = 'bert'
palm2_0 = 'palm2.0'
palm = 'palm-v2'
structbert = 'structbert'
# audio models

2
modelscope/models/nlp/palm_for_text_generation.py
View File

@@ -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):