From: Riceball LEE <redacted>
Date: Fri, 26 Jan 2024 09:10:28 +0000 (+0800)
Subject: gguf : fix "general.alignment" type in gguf_reader.py (#5136)
X-Git-Tag: upstream/0.0.4488~2514
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=38b431de232d1b736b5af19b8c7d72f7075a70bc;p=pkg%2Fggml%2Fsources%2Fllama.cpp

gguf : fix "general.alignment" type in gguf_reader.py (#5136)
---

diff --git a/gguf-py/gguf/gguf_reader.py b/gguf-py/gguf/gguf_reader.py
index 8682765e..5b6d4ba6 100644
--- a/gguf-py/gguf/gguf_reader.py
+++ b/gguf-py/gguf/gguf_reader.py
@@ -107,7 +107,7 @@ 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.UINT32]:
                 raise ValueError('Bad type for general.alignment field')
             self.alignment = new_align.parts[-1][0]
         padding = offs % self.alignment