diff options
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(): |