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

bugfix:retrieve until from until dict

parent 3326c547
......@@ -198,14 +198,14 @@ class GPT3LM(BaseLM):
context_enc = self.tok_encode(context)
inp = context_enc[-(self.max_length - self.max_gen_toks) :]
inps.append(inp)
print(until)
response = oa_completion(
engine=self.engine,
prompt=inps,
max_tokens=self.max_gen_toks,
temperature=0.0,
logprobs=10,
stop=until,
stop=until["until"],
)
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