Skip to content

Commit

Permalink
Fix self.max_tokens in anthropic_llms.py (#1848)
Browse files Browse the repository at this point in the history
Fix bug where `self.max_tokens` was not set
  • Loading branch information
lozhn committed Jun 12, 2024
1 parent 6f43493 commit 793469e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lm_eval/models/anthropic_llms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 793469e

Please sign in to comment.