diff options
| author | Riceball LEE <snowyu.lee@gmail.com> | 2024-01-26 17:10:28 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-26 11:10:28 +0200 |
| commit | 38b431de232d1b736b5af19b8c7d72f7075a70bc (patch) | |
| tree | 6ae5f165fc27bac2ba6a68263b8a2489bd0eda39 /gguf-py | |
| parent | aad0b01d7380a7cdfe0dd42307b18c7b6bac9575 (diff) | |
gguf : fix "general.alignment" type in gguf_reader.py (#5136)
Diffstat (limited to 'gguf-py')
| -rw-r--r-- | gguf-py/gguf/gguf_reader.py | 2 |
1 files changed, 1 insertions, 1 deletions
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 |