mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add types to the encode/decode/token-count endpoints

Browse source
This commit is contained in:
oobabooga 2023-11-07 19:05:36 -08:00
parent f6ca9cfcdc
commit 1b69694fe9
5 changed files with 47 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/llamacpp_model.py
View file

@ -101,7 +101,7 @@ class LlamaCppModel:
return self.model.tokenize(string)
def decode(self, ids):
def decode(self, ids, **kwargs):
return self.model.detokenize(ids).decode('utf-8')
def get_logits(self, tokens):