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

Remove try catch from chat

Browse Source
This commit is contained in:
Mug
2023-04-26 14:38:53 +02:00
parent be2c961bc9
commit 3c130f00ca
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/low_level_api/low_level_api_chat_cpp.py
View File

@@ -356,10 +356,7 @@ n_keep = {self.params.n_keep}
def output(self): def output(self):
self.remaining_tokens = self.params.n_predict self.remaining_tokens = self.params.n_predict
for id in self.generate(): for id in self.generate():
try:
yield llama_cpp.llama_token_to_str(self.ctx, id).decode("utf-8", errors="ignore") yield llama_cpp.llama_token_to_str(self.ctx, id).decode("utf-8", errors="ignore")
except UnicodeDecodeError:
pass
# read user input # read user input
def read_input(self): def read_input(self):