zhangzhicheng.zzc yingda.chen 3 years ago
parent
commit
371530ed1d
3 changed files with 2 additions and 2 deletions
  1. BIN
      data/test/videos/action_recognition_test_video.mp4
  2. +1
    -1
      modelscope/metainfo.py
  3. +1
    -1
      modelscope/models/nlp/palm_for_text_generation.py

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


+ 1
- 1
modelscope/metainfo.py View File

@@ -13,7 +13,7 @@ class Models(object):


# nlp models # nlp models
bert = 'bert' bert = 'bert'
palm2_0 = 'palm2.0'
palm = 'palm-v2'
structbert = 'structbert' structbert = 'structbert'


# audio models # audio models


+ 1
- 1
modelscope/models/nlp/palm_for_text_generation.py View File

@@ -8,7 +8,7 @@ from ..builder import MODELS
__all__ = ['PalmForTextGeneration'] __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): class PalmForTextGeneration(Model):


def __init__(self, model_dir: str, *args, **kwargs): def __init__(self, model_dir: str, *args, **kwargs):


Loading…
Cancel
Save