From 6a13722ca5c07c0b812bdc872c14949d10544234 Mon Sep 17 00:00:00 2001 From: isotr0py <2037008807@qq.com> Date: Tue, 5 Nov 2024 12:42:19 +0800 Subject: [PATCH] code format --- gguf-py/gguf/gguf_reader.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/gguf-py/gguf/gguf_reader.py b/gguf-py/gguf/gguf_reader.py index f0ccc030f..ee94ae056 100644 --- a/gguf-py/gguf/gguf_reader.py +++ b/gguf-py/gguf/gguf_reader.py @@ -6,7 +6,6 @@ from __future__ import annotations import logging import os -import mmap import struct from collections import OrderedDict from typing import Any, Literal, NamedTuple, TypeVar, Union @@ -133,7 +132,7 @@ class GGUFReader: offs += self.alignment - padding self.data_offset = offs self._build_tensors(offs, tensors_fields) - + def __del__(self) -> None: self.data.close() @@ -174,7 +173,7 @@ class GGUFReader: def _get_str(self, offset: int, return_size=False) -> tuple[npt.NDArray[np.uint64], npt.NDArray[np.uint8]]: self.data.seek(offset) slen = struct.unpack('