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
bert = 'bert'
palm2_0 = 'palm2.0'
palm = 'palm-v2'
structbert = 'structbert'

# audio models


+ 1
- 1
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):


Loading…
Cancel
Save