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

Also ignore errors on input prompts

Browse Source
This commit is contained in:
Mug
2023-04-26 14:45:51 +02:00
parent 3c130f00ca
commit 5f81400fcb
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
llama_cpp/llama.py
View File

@@ -358,7 +358,7 @@ class Llama:
if self.verbose:
llama_cpp.llama_reset_timings(self.ctx)
tokens = self.tokenize(input.encode("utf-8"))
tokens = self.tokenize(input.encode("utf-8", errors="ignore"))
self.reset()
self.eval(tokens)
n_tokens = len(tokens)
@@ -416,7 +416,7 @@ class Llama:
completion_tokens: List[llama_cpp.llama_token] = []
# Add blank space to start of prompt to match OG llama tokenizer
prompt_tokens: List[llama_cpp.llama_token] = self.tokenize(
b" " + prompt.encode("utf-8")
b" " + prompt.encode("utf-8", errors="ignore")
)
text: bytes = b""
returned_characters: int = 0
@@ -431,7 +431,7 @@ class Llama:
)
if stop != []:
stop_sequences = [s.encode("utf-8") for s in stop]
stop_sequences = [s.encode("utf-8", errors="ignore") for s in stop]
else:
stop_sequences = []