From 793469e05a3e4cbc0dc8fcffd33df79238949fc8 Mon Sep 17 00:00:00 2001 From: Nikita Lozhnikov Date: Wed, 12 Jun 2024 20:18:34 +0200 Subject: [PATCH] Fix self.max_tokens in anthropic_llms.py (#1848) Fix bug where `self.max_tokens` was not set --- lm_eval/models/anthropic_llms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lm_eval/models/anthropic_llms.py b/lm_eval/models/anthropic_llms.py index 8645aadb02..d011fe074f 100644 --- a/lm_eval/models/anthropic_llms.py +++ b/lm_eval/models/anthropic_llms.py @@ -307,7 +307,7 @@ def __init__( # defaults to os.environ.get("ANTHROPIC_API_KEY") self.client = anthropic.Anthropic() self.temperature = temperature - self.max_token = max_tokens + self.max_tokens = max_tokens self.tokenizer = self.client.get_tokenizer() self.kwargs = kwargs