From ec9450042e73546826b117dbe98b30813123cea9 Mon Sep 17 00:00:00 2001 From: Agus Date: Fri, 5 Jan 2024 14:19:12 +0100 Subject: [PATCH] Update api_key name Co-authored-by: Alvaro Bartolome --- src/distilabel/llm/anyscale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/distilabel/llm/anyscale.py b/src/distilabel/llm/anyscale.py index 2173475f0c..01b3b46674 100644 --- a/src/distilabel/llm/anyscale.py +++ b/src/distilabel/llm/anyscale.py @@ -52,7 +52,7 @@ def __init__( model (str, optional): the model to be used for generation. Defaults to "HuggingFaceH4/zephyr-7b-beta". client (Union[OpenAI, None], optional): an OpenAI client to be used for generation. If `None`, a new client will be created. Defaults to `None`. - openai_api_key (Union[str, None], optional): the OpenAI API key to be used for generation. + api_key (Union[str, None], optional): the Anyscale API key to be used for generation. If `None`, the `OPENAI_API_KEY` environment variable will be used. Defaults to `None`. Visit "https://docs.endpoints.anyscale.com/guides/authenticate/" for more information. max_new_tokens (int, optional): the maximum number of tokens to be generated.