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

irbull · Jan 26, 2024 · 38b431d · 38b431d
1 parent aad0b01
commit 38b431d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gguf-py/gguf/gguf_reader.py b/gguf-py/gguf/gguf_reader.py
@@ -107,7 +107,7 @@ def __init__(self, path: os.PathLike[str] | str, mode: Literal['r' | 'r+' | 'c']
         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