alihan/llama-cpp-python
1
0
Fork 0
mirror of https://github.com/abetlen/llama-cpp-python.git synced 2023-09-07 17:34:22 +03:00
Code Issues Packages Projects Releases Wiki Activity

Cache shared library function calls for static tokens

Browse Source
This commit is contained in:
Andrei Betlen
2023-05-21 19:18:56 -04:00
parent b895511cca
commit cd102e9da1
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
llama_cpp/llama.py
View File

@@ -198,6 +198,8 @@ class Llama:
sorted=sorted,
)
self._candidates = candidates
self._token_nl = Llama.token_nl()
self._token_eos = Llama.token_eos()
def tokenize(self, text: bytes, add_bos: bool = True) -> List[int]:
"""Tokenize a string.
@@ -327,7 +329,7 @@ class Llama:
else last_n_tokens_size
)
logits = self.eval_logits[-1]
nl_logit = logits[Llama.token_nl()]
nl_logit = logits[self._token_nl]
candidates = self._candidates
for i, logit in enumerate(logits):
candidates.data[i].id = llama_cpp.llama_token(i)
@@ -351,7 +353,7 @@ class Llama:
alpha_presence=presence_penalty,
)
if not penalize_nl:
candidates.data[Llama.token_nl()].logit = llama_cpp.c_float(nl_logit)
candidates.data[self._token_nl].logit = llama_cpp.c_float(nl_logit)
if temp.value == 0.0:
return llama_cpp.llama_sample_token_greedy(
ctx=self.ctx,
@@ -688,7 +690,7 @@ class Llama:
presence_penalty=presence_penalty,
repeat_penalty=repeat_penalty,
):
if token == Llama.token_eos():
if token == self._token_eos:
text = self.detokenize(completion_tokens)
finish_reason = "stop"
break