vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gguf-dump: add --data-alignment

Browse source
This commit is contained in:
brian khuu 2024-06-21 21:08:59 +10:00 • committed by Brian
parent c0e6537508
commit de61181079
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
gguf-py/scripts/gguf-dump.py
View file

@ -386,6 +386,7 @@ def main() -> None:
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-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)