Skip to content

Commit

Permalink
inference logging
Browse files Browse the repository at this point in the history
  • Loading branch information
josephjclark committed May 10, 2024
1 parent 3092ccd commit 1334379
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions services/inference/inference.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down

0 comments on commit 1334379

Please sign in to comment.