From df94eb6c125279a9c32bc85de8633371d50afbed Mon Sep 17 00:00:00 2001 From: HsiangNianian Date: Mon, 5 Jan 2026 14:33:10 +0800 Subject: feat: update max_length parameter for TRPGParser and onnx_infer to improve text parsing capabilities --- tests/onnx_infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tests/onnx_infer.py') 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(): -- cgit v1.2.3-70-g09d2