Skip to content

Commit c3c7f04

Browse files
authored
Fix model mapping for AnthropicChat and AnthropicCompletions (#3453)
Previous values for "anthropic-chat" and "anthropic-chat-completions" did not exist as classes. I swapped them out with the correct class, namely "lm_eval.models.anthropic_llms:AnthropicChat"
1 parent a6fd8e6 commit c3c7f04

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lm_eval/models/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323
"""
2424

2525
MODEL_MAPPING = {
26-
"anthropic-chat": "lm_eval.models.anthropic_llms:AnthropicChatLM",
27-
"anthropic-chat-completions": "lm_eval.models.anthropic_llms:AnthropicCompletionsLM",
26+
"anthropic-chat": "lm_eval.models.anthropic_llms:AnthropicChat",
27+
"anthropic-chat-completions": "lm_eval.models.anthropic_llms:AnthropicChat",
2828
"anthropic-completions": "lm_eval.models.anthropic_llms:AnthropicLM",
2929
"dummy": "lm_eval.models.dummy:DummyLM",
3030
"ggml": "lm_eval.models.gguf:GGUFLM",

0 commit comments

Comments
 (0)