From a12df7faf42b3049ec5ec9c4daa8983223cb86a3 Mon Sep 17 00:00:00 2001 From: Bruce MacDonald Date: Tue, 27 Jun 2023 12:32:21 -0400 Subject: [PATCH] fix build --- lib/model_prompts.json => model_prompts.json | 0 ollama.py | 1 - lib/template.py => template.py | 2 +- 3 files changed, 1 insertion(+), 2 deletions(-) rename lib/model_prompts.json => model_prompts.json (100%) rename lib/template.py => template.py (92%) diff --git a/lib/model_prompts.json b/model_prompts.json similarity index 100% rename from lib/model_prompts.json rename to model_prompts.json diff --git a/ollama.py b/ollama.py index 6e8e96ff..cd535e88 100644 --- a/ollama.py +++ b/ollama.py @@ -86,7 +86,6 @@ def unload(model): def generate(model, prompt): # auto load error = load(model) - print(error) if error is not None: return error generated = llms[model]( diff --git a/lib/template.py b/template.py similarity index 92% rename from lib/template.py rename to template.py index 7839508b..e03c2feb 100644 --- a/lib/template.py +++ b/template.py @@ -1,7 +1,7 @@ from difflib import SequenceMatcher import json -with open("model_prompts.json", "r") as f: +with open("./model_prompts.json", "r") as f: model_prompts = json.load(f)