diff --git a/gguf.py b/gguf.py index 1e4f8ea3b..5eb21ee05 100644 --- a/gguf.py +++ b/gguf.py @@ -64,6 +64,7 @@ class GGUFWriter: def __init__(self, fout: IO): self.fout = fout self.offset_tensor = 0 + self.data_alignment = constants.GGUF_DEFAULT_ALIGNMENT self.kv_data = b"" self.kv_data_count = 0 self.ti_data = b"" @@ -191,16 +192,17 @@ class GGUFWriter: dtype = GGMLQuantizationType.F32 if tensor.dtype == np.float32 else GGMLQuantizationType.F16 self.ti_data += struct.pack("