소스 검색

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)
         offs, tensors_fields = self._build_tensors_fields(offs, tensor_count)
         new_align = self.fields.get('general.alignment')
         new_align = self.fields.get('general.alignment')
         if new_align is not None:
         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')
                 raise ValueError('Bad type for general.alignment field')
             self.alignment = new_align.parts[-1][0]
             self.alignment = new_align.parts[-1][0]
         padding = offs % self.alignment
         padding = offs % self.alignment