From 77f52794d43dedd50232212b47ed2970599ca06d Mon Sep 17 00:00:00 2001 From: Jeremy D <115047575+bmosaicml@users.noreply.github.com> Date: Fri, 15 Sep 2023 15:20:30 -0400 Subject: [PATCH] Update llmfoundry/models/inference_api_wrapper/interface.py Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com> --- llmfoundry/models/inference_api_wrapper/interface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/models/inference_api_wrapper/interface.py b/llmfoundry/models/inference_api_wrapper/interface.py index a513654c18..433d4d8e06 100644 --- a/llmfoundry/models/inference_api_wrapper/interface.py +++ b/llmfoundry/models/inference_api_wrapper/interface.py @@ -41,7 +41,7 @@ def __init__(self, model_cfg: Dict, tokenizer: AutoTokenizer): def get_metrics(self, is_train: bool = False): if is_train: raise NotImplementedError( - 'You cannot use OpenAI inference wrappers for training') + 'You cannot use inference wrappers for training') else: metrics = self.eval_metrics