vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

gguf.py : support any type

Browse source
This commit is contained in:
klosax 2023-07-29 21:29:07 +02:00 • committed by GitHub
parent 2c22e3bcdb
commit 9577821487
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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

@ -270,10 +270,10 @@ class GGUFWriter:
def write_tokenizer_model(self, model: str):
self.write_string(constants.KEY_TOKENIZER_MODEL, model)
def write_token_list(self, tokens: List[str]):
def write_token_list(self, tokens: List):
self.write_array(constants.KEY_TOKENIZER_LIST, tokens)
def write_token_scores(self, scores: List[float]):
def write_token_scores(self, scores: List:
self.write_array(constants.KEY_TOKENIZER_SCORES, scores)