Correct error message for inference wrapper (#1459)

mosaicml · Aug 18, 2024 · b81af3a · b81af3a
1 parent d413e01
commit b81af3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/models/inference_api_wrapper/openai_causal_lm.py b/llmfoundry/models/inference_api_wrapper/openai_causal_lm.py
@@ -102,7 +102,7 @@ def try_generate_completion(self, prompt: str, num_tokens: int):
                 break
             except RateLimitError as e:
                 if 'You exceeded your current quota' in str(
-                    e._message,
+                    e.message,
                 ):  # pyright: ignore
                     raise e
                 delay *= 2 * (1 + random.random())