Skip to content

Commit a69febd

Browse files
committed
gguf-dump.py: Add tensor overview count
1 parent b000526 commit a69febd

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

gguf-py/scripts/gguf-dump.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -227,8 +227,11 @@ def dump_markdown_metadata(reader: GGUFReader, args: argparse.Namespace) -> None
227227
tensor_name_to_key[tensor.name] = key
228228
tensor_groups[tensor_prefix].append(tensor)
229229

230-
# Generate Markdown metadata
231-
markdown_content += "## Tensor Groups\n"
230+
# Tensors Mapping Dump
231+
markdown_content += f'## Tensors Overview {element_count_rounded_notation(total_elements)} Elements\n'
232+
markdown_content += f'Total number of elements in all tensors: {total_elements} Elements\n'
233+
markdown_content += '\n'
234+
232235
for group in tensor_prefix_order:
233236
tensors = tensor_groups[group]
234237
group_elements = sum(tensor.n_elements for tensor in tensors)

0 commit comments

Comments
 (0)