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

fix mode

Browse source
This commit is contained in:
isotr0py 2024-11-04 16:06:46 +08:00
parent bcef54e10a
commit 205676ceb7
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -88,8 +88,9 @@ class GGUFReader:
} }
def __init__(self, path: os.PathLike[str] | str, mode: Literal['r', 'r+', 'c'] = 'r'): def __init__(self, path: os.PathLike[str] | str, mode: Literal['r', 'r+', 'c'] = 'r'):
self.data = open(path, mode="rb") file_mode = "rb" if mode == 'r' else 'rb+'
self.mmap = np.memmap(path, mode = mode) self.data = open(path, mode=file_mode)
self.mmap = np.memmap(self.data, mode = mode)
offs = 0 offs = 0
# Check for GGUF magic # Check for GGUF magic
@ -129,7 +130,7 @@ class GGUFReader:
if padding != 0: if padding != 0:
offs += self.alignment - padding offs += self.alignment - padding
self.data_offset = offs self.data_offset = offs
# self._build_tensors(offs, tensors_fields) self._build_tensors(offs, tensors_fields)
self.data.close() self.data.close()
_DT = TypeVar('_DT', bound = npt.DTypeLike) _DT = TypeVar('_DT', bound = npt.DTypeLike)