From 1334379edf5e2a6528c3db4d24d23c157a3f2c17 Mon Sep 17 00:00:00 2001 From: Joe Clark Date: Fri, 10 May 2024 14:25:10 +0100 Subject: [PATCH] inference logging --- services/inference/inference.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/services/inference/inference.py b/services/inference/inference.py index ae46f72..09934d4 100644 --- a/services/inference/inference.py +++ b/services/inference/inference.py @@ -1,12 +1,11 @@ -import logging - from .models.gpt3_turbo import generate as gpt3_turbo from .models.codet5 import generate as codet5 from .models.gpt_ft import generate as gpt_ft from util import DictObj -logging.basicConfig(level=logging.INFO) -logger = logging.getLogger(__name__) +from util import createLogger + +logger = createLogger("inference.prompts") # all http calls to inference need a model and prompt, and optionally an args dict