Skip to content

Commit 9ab2f4a

Browse files
authored
apply newbyteorder fixes
1 parent e4727e3 commit 9ab2f4a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

gguf-py/gguf/gguf_reader.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def __init__(self, path: os.PathLike[str] | str, mode: Literal['r', 'r+', 'c'] =
144144
# If we get 0 here that means it's (probably) a GGUF file created for
145145
# the opposite byte order of the machine this script is running on.
146146
self.byte_order = 'S'
147-
temp_version = temp_version.newbyteorder(self.byte_order)
147+
temp_version = temp_version.view(temp_version.dtype.newbyteorder(self.byte_order))
148148
version = temp_version[0]
149149
if version not in READER_SUPPORTED_VERSIONS:
150150
raise ValueError(f'Sorry, file appears to be version {version} which we cannot handle')
@@ -198,9 +198,7 @@ def _get(
198198
itemsize = int(np.empty([], dtype = dtype).itemsize)
199199
end_offs = offset + itemsize * count
200200
arr = self.data[offset:end_offs].view(dtype=dtype)[:count]
201-
if override_order is None:
202-
return arr
203-
return arr.view(arr.dtype.newbyteorder(override_order))
201+
return arr.view(arr.dtype.newbyteorder(self.byte_order if override_order is None else override_order))
204202

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

0 commit comments

Comments
 (0)