From 66023bded9da40dbf0752f3934a9fef144badf51 Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Mon, 20 May 2024 19:05:12 +0000 Subject: [PATCH] Fix tensor size limit display --- src/model_explorer_onnx/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model_explorer_onnx/main.py b/src/model_explorer_onnx/main.py index bcd3b55..bf232a3 100644 --- a/src/model_explorer_onnx/main.py +++ b/src/model_explorer_onnx/main.py @@ -58,7 +58,7 @@ def display_tensor_json(tensor: ir.TensorProtocol | None, settings: Settings) -> try: array = _tensor_to_numpy(tensor) size_limit = settings.const_element_count_limit - if size_limit < 0 or size_limit > array.size: + if size_limit < 0 or size_limit >= array.size: # Use separators=(',', ':') to remove spaces return json.dumps(array.tolist(), separators=(",", ":")) # Show the first `size_limit` elements if the tensor is too large