Explorar el Código

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

Jett Janiak hace 1 año
padre
commit
4601a8bb67
Se han modificado 1 ficheros con 4 adiciones y 5 borrados
  1. 4 5
      gguf-py/gguf/gguf_reader.py

+ 4 - 5
gguf-py/gguf/gguf_reader.py

@@ -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: