Update llmfoundry/models/inference_api_wrapper/interface.py

Co-authored-by: Daniel King <[email protected]>
mosaicml · Sep 15, 2023 · 77f5279 · 77f5279
1 parent 2bd933a
commit 77f5279
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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