Skip to content

GitLab

Commit 1815286c authored by Leo Gao's avatar Leo Gao
Browse files

Add missing import

parent 5f4c7c50
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
lm_eval/models/gpt3.py
View file @ 1815286c
...@@ -49,6 +49,7 @@ class GPT3LM(LM): ...@@ -49,6 +49,7 @@ class GPT3LM(LM):
return cls(engine=args.get("engine", "davinci")) return cls(engine=args.get("engine", "davinci"))
def loglikelihood(self, requests): def loglikelihood(self, requests):
import openai
for chunk in tqdm(utils.chunks(requests, self.REQ_CHUNK_SIZE)): for chunk in tqdm(utils.chunks(requests, self.REQ_CHUNK_SIZE)):
inps = [] inps = []
ctxlens = [] ctxlens = []
......
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