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

gguf-py: gguf-dump: Respect --no-tensor flag in JSON mode.

Browse source
This commit is contained in:
KerfuffleV2 2023-11-11 13:15:27 -07:00
parent e86fc56f75
commit 0c201b9142
1 changed files with 8 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -86,6 +86,7 @@ def dump_metadata_json(reader: GGUFReader, args: argparse.Namespace) -> None:
curr["value"] = str(bytes(field.parts[-1]), encoding="utf-8") curr["value"] = str(bytes(field.parts[-1]), encoding="utf-8")
else: else:
curr["value"] = field.parts[-1].tolist()[0] curr["value"] = field.parts[-1].tolist()[0]
if not args.no_tensors:
for idx, tensor in enumerate(reader.tensors): for idx, tensor in enumerate(reader.tensors):
tensors[tensor.name] = { tensors[tensor.name] = {
"index": idx, "index": idx,