From 9577821487909aac1ee33654ac0e9ade9a63f886 Mon Sep 17 00:00:00 2001 From: klosax <131523366+klosax@users.noreply.github.com> Date: Sat, 29 Jul 2023 21:29:07 +0200 Subject: [PATCH] gguf.py : support any type --- gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/gguf.py b/gguf.py index 88d7e43fb..7e15e5a77 100644 --- a/gguf.py +++ b/gguf.py @@ -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)