diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/base_model_trpgner/inference/__init__.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/base_model_trpgner/inference/__init__.py b/src/base_model_trpgner/inference/__init__.py index 41f3504..5824666 100644 --- a/src/base_model_trpgner/inference/__init__.py +++ b/src/base_model_trpgner/inference/__init__.py @@ -232,6 +232,9 @@ class TRPGParser: >>> result['metadata']['speaker'] '风雨' """ + if max_length > 512: + raise ValueError("max_length 不能超过 512") + # Tokenize inputs = self.tokenizer( text, |