fix: Fixes wrong input type for raw_dtype in ggml to gguf scripts

This commit is contained in:
farbod 2024-08-10 21:46:00 +03:30
parent 1e6f6554aa
commit 66a4225825

View file

@ -116,7 +116,7 @@ class Tensor:
assert quant is not None, 'Unknown tensor type'
(blksize, tysize) = quant
offset += 12
self.dtype= dtype
self.dtype= gguf.GGMLQuantizationType(dtype)
self.dims = struct.unpack(f'<{n_dims}I', data[offset:offset + (4 * n_dims)])
offset += 4 * n_dims
self.name = bytes(data[offset:offset + name_len])