Skip to content

Commit ca34531

Browse files
jettjaniakmglambda
authored andcommitted
gguf-py : numpy 2 newbyteorder fix (ggml-org#9772)
1 parent 599f7b2 commit ca34531

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

gguf-py/gguf/gguf_reader.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -145,11 +145,10 @@ def _get(
145145
count = int(count)
146146
itemsize = int(np.empty([], dtype = dtype).itemsize)
147147
end_offs = offset + itemsize * count
148-
return (
149-
self.data[offset:end_offs]
150-
.view(dtype = dtype)[:count]
151-
.newbyteorder(override_order or self.byte_order)
152-
)
148+
arr = self.data[offset:end_offs].view(dtype=dtype)[:count]
149+
if override_order is None:
150+
return arr
151+
return arr.view(arr.dtype.newbyteorder(override_order))
153152

154153
def _push_field(self, field: ReaderField, skip_sum: bool = False) -> int:
155154
if field.name in self.fields:

0 commit comments

Comments
 (0)