baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix build

Browse source
This commit is contained in:
Bruce MacDonald 2023-06-27 12:32:21 -04:00
parent 6599268d78
commit a12df7faf4
3 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
lib/model_prompts.json → model_prompts.json
View file

1
ollama.py
View file

@ -86,7 +86,6 @@ def unload(model):
def generate(model, prompt): def generate(model, prompt):
# auto load # auto load
error = load(model) error = load(model)
print(error)
if error is not None: if error is not None:
return error return error
generated = llms[model]( generated = llms[model](

2
lib/template.py → template.py
View file

@ -1,7 +1,7 @@
from difflib import SequenceMatcher from difflib import SequenceMatcher
import json import json
with open("model_prompts.json", "r") as f: with open("./model_prompts.json", "r") as f:
model_prompts = json.load(f) model_prompts = json.load(f)