Skip to content

Commit

Permalink
OpenAI Completions -- fix passing of unexpected 'until' arg (Eleuther…
Browse files Browse the repository at this point in the history
  • Loading branch information
haileyschoelkopf authored and jmercat committed Sep 25, 2024
1 parent f70915f commit 77645c4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lm_eval/models/openai_completions.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,7 +281,7 @@ def sameuntil_chunks(xs, size):
**{
k: v
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):
Expand Down

0 comments on commit 77645c4

Please sign in to comment.