gguf.py : support any type

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

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)