浏览代码

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

Riceball LEE 2 年之前
父节点
当前提交
38b431de23
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      gguf-py/gguf/gguf_reader.py

+ 1 - 1
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