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

updated

Browse Source
This commit is contained in:
Yufeng
2022-11-15 16:38:16 +08:00
parent c3457ce41d
commit eea3f919ec
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/models/nlp/txl_poem/fastpoem.py
View File

@@ -16,7 +16,6 @@ import torch.nn.functional as F
from pypinyin import FINALS, FINALS_TONE, TONE3, pinyin
from .arguments import get_args
from .gpt2 import mpu
from .gpt2.configure_data import configure_data
from .gpt2.data_utils import make_tokenizer
@@ -28,11 +27,12 @@ from .gpt2.utils import (Timers, get_checkpoint_iteration, load_checkpoint,
open_old_pronounce = 1
class APIException(Exception):
def __init__(self, message):
super().__init__(message)
class CanNotReturnException(APIException):
@@ -40,12 +40,14 @@ class CanNotReturnException(APIException):
self.payload = payload
super().__init__(message)
class InputTooLongException(APIException):
def __init__(self, message, payload=None):
self.payload = payload
super().__init__(message)
def get_model(args):
"""Build the model."""