diff options
| author |   HsiangNianian <i@jyunko.cn> | 2026-01-05 14:33:10 +0800 |
|---|---|---|
| committer | HsiangNianian <i@jyunko.cn> | 2026-01-05 14:33:10 +0800 |
| commit | df94eb6c125279a9c32bc85de8633371d50afbed (patch) | |
| tree | fa7e99e5078bbcb62e7b9dcc181f4fbc282129c4 /tests/onnx_infer.py | |
| parent | 25380fb4de77966a0f3d00681be25857c27b0869 (diff) | |
| download | base-model-df94eb6c125279a9c32bc85de8633371d50afbed.tar.gz base-model-df94eb6c125279a9c32bc85de8633371d50afbed.zip | |
feat: update max_length parameter for TRPGParser and onnx_infer to improve text parsing capabilities
Diffstat (limited to 'tests/onnx_infer.py')
| -rw-r--r-- | tests/onnx_infer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/onnx_infer.py b/tests/onnx_infer.py index 4ffca25..f4c7f9d 100644 --- a/tests/onnx_infer.py +++ b/tests/onnx_infer.py @@ -19,7 +19,7 @@ def softmax(x): return e / e.sum(axis=-1, keepdims=True) text = sys.argv[1] if len(sys.argv) > 1 else "风雨 2024-06-08 21:44:59 剧烈的疼痛..." -inputs = tok(text, return_tensors="np", return_offsets_mapping=True, padding="max_length", truncation=True, max_length=128) +inputs = tok(text, return_tensors="np", return_offsets_mapping=True, padding="max_length", truncation=True, max_length=512) feed = {} for inp in sess.get_inputs(): |