From eea3f919ecd8421a5b1ba663cf180b66c3a8858a Mon Sep 17 00:00:00 2001 From: Yufeng Date: Tue, 15 Nov 2022 16:38:16 +0800 Subject: [PATCH] updated --- modelscope/models/nlp/txl_poem/fastpoem.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modelscope/models/nlp/txl_poem/fastpoem.py b/modelscope/models/nlp/txl_poem/fastpoem.py index b90fdaa3..822d62e0 100755 --- a/modelscope/models/nlp/txl_poem/fastpoem.py +++ b/modelscope/models/nlp/txl_poem/fastpoem.py @@ -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."""