From a69febd886cb5e8ee5018e91ae559d6a213a42d1 Mon Sep 17 00:00:00 2001 From: brian khuu Date: Mon, 10 Jun 2024 22:30:41 +1000 Subject: [PATCH] gguf-dump.py: Add tensor overview count --- gguf-py/scripts/gguf-dump.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/gguf-py/scripts/gguf-dump.py b/gguf-py/scripts/gguf-dump.py index 2e338ab49..73e08a9fd 100755 --- a/gguf-py/scripts/gguf-dump.py +++ b/gguf-py/scripts/gguf-dump.py @@ -227,8 +227,11 @@ def dump_markdown_metadata(reader: GGUFReader, args: argparse.Namespace) -> None tensor_name_to_key[tensor.name] = key tensor_groups[tensor_prefix].append(tensor) - # Generate Markdown metadata - markdown_content += "## Tensor Groups\n" + # Tensors Mapping Dump + markdown_content += f'## Tensors Overview {element_count_rounded_notation(total_elements)} Elements\n' + markdown_content += f'Total number of elements in all tensors: {total_elements} Elements\n' + markdown_content += '\n' + for group in tensor_prefix_order: tensors = tensor_groups[group] group_elements = sum(tensor.n_elements for tensor in tensors)