hueso/Alpaca
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added exception catch to Ollama's output in case it can't decode message

Browse Source
This commit is contained in:
jeffser 2024-08-19 23:06:58 -06:00
parent ff927d6c77
commit ed54b2846a
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/local_instance.py
View File

@ -19,10 +19,13 @@ overrides = {}
def log_output(pipe): def log_output(pipe):
with open(os.path.join(data_dir, 'tmp.log'), 'a') as f: with open(os.path.join(data_dir, 'tmp.log'), 'a') as f:
with pipe: with pipe:
for line in iter(pipe.readline, ''): try:
print(line, end='') for line in iter(pipe.readline, ''):
f.write(line) print(line, end='')
f.flush() f.write(line)
f.flush()
except:
pass
def start(): def start():
if not os.path.isdir(os.path.join(cache_dir, 'tmp/ollama')): if not os.path.isdir(os.path.join(cache_dir, 'tmp/ollama')):