Skip to content

Commit

Permalink
gguf-dump: add --data-alignment
Browse files Browse the repository at this point in the history
  • Loading branch information
mofosyne committed Jun 21, 2024
1 parent 3ef9efc commit 86040f8
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions gguf-py/scripts/gguf-dump.py
Original file line number Diff line number Diff line change
Expand Up @@ -385,15 +385,16 @@ 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")

args = parser.parse_args(None if len(sys.argv) > 1 else ["--help"])

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')
Expand All @@ -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)

Expand Down

0 comments on commit 86040f8

Please sign in to comment.