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

gguf-py : reader prints warnings on duplicate keys

Browse source
This commit is contained in:
Georgi Gerganov 2024-04-26 14:32:22 +03:00
parent e1b2bf783e
commit ed42711b90
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
gguf-py/gguf/gguf_reader.py
View file

@ -139,7 +139,12 @@ class GGUFReader:
def _push_field(self, field: ReaderField, skip_sum: bool = False) -> int:
if field.name in self.fields:
raise KeyError(f'Duplicate {field.name} already in list at offset {field.offset}')
# TODO: add option to generate error on duplicate keys
#raise KeyError(f'Duplicate {field.name} already in list at offset {field.offset}')
print(f'Warning: Duplicate key {field.name} at offset {field.offset}')
self.fields[field.name + '_{}'.format(field.offset)] = field
else:
self.fields[field.name] = field
return 0 if skip_sum else sum(int(part.nbytes) for part in field.parts)