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

revert unnecessary change

Browse source
This commit is contained in:
isotr0py 2024-11-05 01:48:31 +08:00
parent a92c920eec
commit ad6fd8de25
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -152,10 +152,10 @@ class GGUFReader:
) -> npt.NDArray[Any]: ) -> npt.NDArray[Any]:
count = int(count) count = int(count)
itemsize = np.dtype(dtype).itemsize itemsize = np.dtype(dtype).itemsize
new_offset = offset + itemsize * count end_offs = offset + itemsize * count
self.data.seek(new_offset) self.data.seek(end_offs)
return ( return (
self.mmap[offset:new_offset] self.mmap[offset:end_offs]
.view(dtype = dtype)[:count] .view(dtype = dtype)[:count]
.newbyteorder(override_order or self.byte_order) .newbyteorder(override_order or self.byte_order)
) )