Browse Source

Merge remote-tracking branch 'origin' into feat/fill_mask

master
suluyan 3 years ago
parent
commit
e6bb781c6d
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      modelscope/metainfo.py
  2. +1
    -1
      modelscope/models/nlp/palm_for_text_generation.py

+ 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