From de6118107915b81b34dec5bc5f20722dce963ab3 Mon Sep 17 00:00:00 2001 From: brian khuu Date: Fri, 21 Jun 2024 21:08:59 +1000 Subject: [PATCH] gguf-dump: add --data-alignment --- gguf-py/scripts/gguf-dump.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/gguf-py/scripts/gguf-dump.py b/gguf-py/scripts/gguf-dump.py index 8b39cc56a..e4a084415 100755 --- a/gguf-py/scripts/gguf-dump.py +++ b/gguf-py/scripts/gguf-dump.py @@ -385,7 +385,8 @@ def main() -> None: parser.add_argument("--no-tensors", action="store_true", help="Don't dump tensor metadata") parser.add_argument("--json", action="store_true", help="Produce JSON output") parser.add_argument("--json-array", action="store_true", help="Include full array values in JSON output (long)") - parser.add_argument("--data-offset", action="store_true", help="Start of data offset") + parser.add_argument("--data-offset", action="store_true", help="Start of data offset") + parser.add_argument("--data-alignment", action="store_true", help="Data alignment applied globally to data field") parser.add_argument("--markdown", action="store_true", help="Produce markdown output") parser.add_argument("--verbose", action="store_true", help="increase output verbosity") @@ -393,7 +394,7 @@ def main() -> None: logging.basicConfig(level=logging.DEBUG if args.verbose else logging.INFO) - if not args.json and not args.markdown and not args.data_offset: + if not args.json and not args.markdown and not args.data_offset and not args.data_alignment: logger.info(f'* Loading: {args.model}') reader = GGUFReader(args.model, 'r') @@ -403,7 +404,9 @@ def main() -> None: elif args.markdown: dump_markdown_metadata(reader, args) elif args.data_offset: - print(reader.start_data_offset) + print(reader.start_data_offset) # noqa: NP100 + elif args.data_alignment: + print(reader.alignment) # noqa: NP100 else: dump_metadata(reader, args)