From 205676ceb7855342741242c9742dd2ae257bb812 Mon Sep 17 00:00:00 2001 From: isotr0py <2037008807@qq.com> Date: Mon, 4 Nov 2024 16:06:46 +0800 Subject: [PATCH] fix mode --- gguf-py/gguf/gguf_reader.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/gguf-py/gguf/gguf_reader.py b/gguf-py/gguf/gguf_reader.py index 3dd004d66..fd32a3d1f 100644 --- a/gguf-py/gguf/gguf_reader.py +++ b/gguf-py/gguf/gguf_reader.py @@ -88,8 +88,9 @@ class GGUFReader: } def __init__(self, path: os.PathLike[str] | str, mode: Literal['r', 'r+', 'c'] = 'r'): - self.data = open(path, mode="rb") - self.mmap = np.memmap(path, mode = mode) + file_mode = "rb" if mode == 'r' else 'rb+' + self.data = open(path, mode=file_mode) + self.mmap = np.memmap(self.data, mode = mode) offs = 0 # Check for GGUF magic @@ -129,7 +130,7 @@ class GGUFReader: if padding != 0: offs += self.alignment - padding self.data_offset = offs - # self._build_tensors(offs, tensors_fields) + self._build_tensors(offs, tensors_fields) self.data.close() _DT = TypeVar('_DT', bound = npt.DTypeLike)