Skip to content

GitLab

Commit 701bd09f authored by baberabb's avatar baberabb
Browse files

bugfix:retrieve until from until dict

parent 3326c547
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
lm_eval/models/gpt3.py
View file @ 701bd09f
...@@ -198,14 +198,14 @@ class GPT3LM(BaseLM): ...@@ -198,14 +198,14 @@ class GPT3LM(BaseLM):
context_enc = self.tok_encode(context) context_enc = self.tok_encode(context)
inp = context_enc[-(self.max_length - self.max_gen_toks) :] inp = context_enc[-(self.max_length - self.max_gen_toks) :]
inps.append(inp) inps.append(inp)
print(until)
response = oa_completion( response = oa_completion(
engine=self.engine, engine=self.engine,
prompt=inps, prompt=inps,
max_tokens=self.max_gen_toks, max_tokens=self.max_gen_toks,
temperature=0.0, temperature=0.0,
logprobs=10, logprobs=10,
stop=until, stop=until["until"],
) )
for resp, (context, until_) in zip(response.choices, chunk): for resp, (context, until_) 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