]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
gguf-py : numpy 2 newbyteorder fix (#9772)
authorJett Janiak <redacted>
Fri, 13 Dec 2024 14:48:44 +0000 (15:48 +0100)
committerGitHub <redacted>
Fri, 13 Dec 2024 14:48:44 +0000 (16:48 +0200)
gguf-py/gguf/gguf_reader.py

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