Mirrors/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2025-01-06 02:48:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

gguf : fix set is not subscriptable

Browse Source
This commit is contained in:
M. Yusuf Sarıgöz 2023-07-30 01:01:13 +03:00
parent 06c3e4a1a7
commit 32e037ffbe
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gguf.py
View File

@ -149,7 +149,7 @@ class GGUFWriter:
elif vtype == GGUFValueType.ARRAY:
ltype = set([GGUFValueType.get_type(item) for item in val])
assert len(ltype) == 1, "All items in a GGUF array should be of the same type"
self.fout.write(struct.pack("<I", ltype[0]))
self.fout.write(struct.pack("<I", list(ltype)[0]))
self.fout.write(struct.pack("<I", len(val)))
for item in val:
self.write_val(item, write_vtype=False)
@ -273,7 +273,7 @@ class GGUFWriter:
def write_token_list(self, tokens: List):
self.write_array(constants.KEY_TOKENIZER_LIST, tokens)
def write_token_scores(self, scores: List:
def write_token_scores(self, scores: List[float]):
self.write_array(constants.KEY_TOKENIZER_SCORES, scores)