CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-11-04 09:32:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

gguf : handle already encoded string

Browse Source
This commit is contained in:
M. Yusuf Sarıgöz
2023-07-29 19:56:06 +03:00
parent 8ad7cd49fb
commit 0f5e57f01d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gguf.py
View File

@@ -45,7 +45,7 @@ class GGUFValueType(IntEnum):
@staticmethod @staticmethod
def get_type(val): def get_type(val):
if isinstance(val, str): if isinstance(val, str) or isinstance(val, bytes):
return GGUFValueType.STRING return GGUFValueType.STRING
elif isinstance(val, list): elif isinstance(val, list):
return GGUFValueType.ARRAY return GGUFValueType.ARRAY
@@ -143,7 +143,7 @@ class GGUFWriter:
elif vtype == GGUFValueType.BOOL: elif vtype == GGUFValueType.BOOL:
self.fout.write(struct.pack("?", val)) self.fout.write(struct.pack("?", val))
elif vtype == GGUFValueType.STRING: elif vtype == GGUFValueType.STRING:
encoded_val = val.encode("utf8") encoded_val = val.encode("utf8") if isinstance(val, str) else val
self.fout.write(struct.pack("<I", len(encoded_val))) self.fout.write(struct.pack("<I", len(encoded_val)))
self.fout.write(encoded_val) self.fout.write(encoded_val)
elif vtype == GGUFValueType.ARRAY: elif vtype == GGUFValueType.ARRAY: