From eda5614f415ed1a0030eccd97a6e6e7851e8496e Mon Sep 17 00:00:00 2001 From: jay-johnson Date: Mon, 13 Nov 2023 17:44:30 +0000 Subject: [PATCH] gguf-py readme example fixes for keys: general.architecture and general.alignment --- gguf-py/examples/writer.py | 1 - gguf-py/gguf/gguf_reader.py | 8 +++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/gguf-py/examples/writer.py b/gguf-py/examples/writer.py index f39eed1af..731873a7d 100755 --- a/gguf-py/examples/writer.py +++ b/gguf-py/examples/writer.py @@ -15,7 +15,6 @@ def writer_example() -> None: # Example usage with a file gguf_writer = GGUFWriter("example.gguf", "llama") - gguf_writer.add_architecture() gguf_writer.add_block_count(12) gguf_writer.add_uint32("answer", 42) # Write a 32-bit integer gguf_writer.add_float32("answer_in_float", 42.0) # Write a 32-bit float diff --git a/gguf-py/gguf/gguf_reader.py b/gguf-py/gguf/gguf_reader.py index 8682765ed..eaef19a74 100644 --- a/gguf-py/gguf/gguf_reader.py +++ b/gguf-py/gguf/gguf_reader.py @@ -107,7 +107,13 @@ class GGUFReader: offs, tensors_fields = self._build_tensors_fields(offs, tensor_count) new_align = self.fields.get('general.alignment') if new_align is not None: - if new_align.types != [GGUFValueType.UINT64]: + if ( + ( + new_align.types != [GGUFValueType.UINT64] + ) and ( + new_align.types != [GGUFValueType.UINT32] + ) + ): raise ValueError('Bad type for general.alignment field') self.alignment = new_align.parts[-1][0] padding = offs % self.alignment