diff options
| author |   HsiangNianian <i@jyunko.cn> | 2026-01-05 14:33:37 +0800 |
|---|---|---|
| committer | HsiangNianian <i@jyunko.cn> | 2026-01-05 14:33:37 +0800 |
| commit | 753119374a952cac55a0d87c13e5ae081e09de4b (patch) | |
| tree | f0cd05d84ec9e4b3b47dec425ce80275dcaa214b /src/base_model_trpgner/inference/__init__.py | |
| parent | df94eb6c125279a9c32bc85de8633371d50afbed (diff) | |
| download | base-model-753119374a952cac55a0d87c13e5ae081e09de4b.tar.gz base-model-753119374a952cac55a0d87c13e5ae081e09de4b.zip | |
feat: add max_length validation in TRPGParser to enforce input constraints
Diffstat (limited to 'src/base_model_trpgner/inference/__init__.py')
| -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, |