From 46a9a0656af501a58bf4e9ad000f134502859490 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sigbj=C3=B8rn=20Skj=C3=A6ret?= Date: Sun, 22 Feb 2026 10:38:58 +0100 Subject: [PATCH] enforce proper alignment in add_custom_alignment --- gguf-py/gguf/gguf_writer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py index 4245d18bc4..9ee3ac9e8f 100644 --- a/gguf-py/gguf/gguf_writer.py +++ b/gguf-py/gguf/gguf_writer.py @@ -501,6 +501,8 @@ class GGUFWriter: self.add_uint32(Keys.General.QUANTIZATION_VERSION, quantization_version) def add_custom_alignment(self, alignment: int) -> None: + if alignment <= 0 or (alignment & (alignment - 1)) != 0: + raise ValueError('Invalid alignment: must be a non-zero power of two') self.data_alignment = alignment self.add_uint32(Keys.General.ALIGNMENT, alignment)