Skip to content

GitLab

Unverified Commit 34c9b7e4 authored by Hailey Schoelkopf's avatar Hailey Schoelkopf Committed by GitHub
Browse files

OpenAI Completions -- fix passing of unexpected 'until' arg (#1612)

parent 65546905
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/openai_completions.py
View file @ 34c9b7e4
...@@ -281,7 +281,7 @@ class OpenaiCompletionsLM(TemplateLM): ...@@ -281,7 +281,7 @@ class OpenaiCompletionsLM(TemplateLM):
**{ **{
k: v k: v
for k, v in request_args.items() for k, v in request_args.items()
if k not in ["do_sample", "max_gen_toks"] if k not in {"do_sample", "max_gen_toks", "until"}
}, },
) )
for resp, (context, args_) in zip(response.choices, chunk): for resp, (context, args_) in zip(response.choices, chunk):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment