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