change to `API_KEY` for local completions

686098d0 · Baber · 31833bf8 · 686098d0
Commit 686098d0 authored Nov 21, 2024 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/models/openai_completions.py lm_eval/models/openai_completions.py +1 -1

No files found.
--- a/lm_eval/models/openai_completions.py
+++ b/lm_eval/models/openai_completions.py
@@ -91,7 +91,7 @@ class LocalCompletionsAPI(TemplateAPI):

    @property
    def api_key(self):
-        return os.environ.get("OPENAI_API_KEY", "")
+        return os.environ.get("API_KEY", "")


 @register_model("local-chat-completions")