From a234bf821b8cad89b7318efed495ad43da5988e3 Mon Sep 17 00:00:00 2001 From: Christian Zhou-Zheng Date: Sun, 9 Jun 2024 11:23:55 -0400 Subject: [PATCH] fix linting --- convert-hf-to-gguf.py | 10 ++++------ gguf-py/gguf/gguf_writer.py | 9 ++++----- 2 files changed, 8 insertions(+), 11 deletions(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index de8e10813..271ded8c4 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2891,15 +2891,13 @@ def main() -> None: model_instance.gguf_writer.add_quantization_version(gguf.GGML_QUANT_VERSION) if args.vocab_only: - logger.info(f"Exporting model vocab...") + logger.info("Exporting model vocab...") model_instance.write_vocab() - logger.info(f"Model vocab successfully exported.") + logger.info("Model vocab successfully exported.") else: - logger.info(f"Exporting model...") + logger.info("Exporting model...") model_instance.write() - logger.info(f"Model successfully exported.") - - + logger.info("Model successfully exported.") if __name__ == '__main__': main() diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 4b1025820..766ae86b4 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -30,7 +30,6 @@ from .constants import ( ) from .quants import quant_shape_from_byte_shape -from .constants import Keys logger = logging.getLogger(__name__) @@ -191,7 +190,7 @@ class GGUFWriter: def add_shard_kv_data(self) -> None: if self.split_arguments.split_style == SplitStyle.NONE: return - + total_tensors = sum(len(t) for t in self.tensors) for i in range(len(self.fout)): # just see whether it exists @@ -746,11 +745,11 @@ class GGUFWriter: return tensor.data_type.elements_to_bytes(np.prod(tensor.shape)) except AttributeError: # numpy ndarray[Any, Any] return tensor.nbytes - + @staticmethod def get_tensors_total_size(tensors) -> int: return sum(GGUFWriter.get_tensor_size(ti) for ti in tensors) - + @staticmethod def split_str_to_n_bytes(split_str: str) -> int: if split_str.endswith("K"): @@ -778,4 +777,4 @@ class GGUFWriter: if abs(fnum) < 1000.0: return f"{fnum:3.1f}{unit}" fnum /= 1000.0 - return f"{fnum:.1f}T - over 1TB, --split recommended" \ No newline at end of file + return f"{fnum:.1f}T - over 1TB, --split recommended"