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

gguf-py : numpy 2 newbyteorder fix (#9772)

Browse source
This commit is contained in:
Jett Janiak 2024-12-13 15:48:44 +01:00 • committed by GitHub
parent 9f35e44592
commit 4601a8bb67
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -145,11 +145,10 @@ class GGUFReader:
count = int(count) count = int(count)
itemsize = int(np.empty([], dtype = dtype).itemsize) itemsize = int(np.empty([], dtype = dtype).itemsize)
end_offs = offset + itemsize * count end_offs = offset + itemsize * count
return ( arr = self.data[offset:end_offs].view(dtype=dtype)[:count]
self.data[offset:end_offs] if override_order is None:
.view(dtype = dtype)[:count] return arr
.newbyteorder(override_order or self.byte_order) return arr.view(arr.dtype.newbyteorder(override_order))
)
def _push_field(self, field: ReaderField, skip_sum: bool = False) -> int: def _push_field(self, field: ReaderField, skip_sum: bool = False) -> int:
if field.name in self.fields: if field.name in self.fields: