Guangsheng Bao commited on
Commit
8b5ebbb
·
1 Parent(s): fdd6aa7

minor fix

Browse files
Files changed (1) hide show
  1. glimpse.py +3 -3
glimpse.py CHANGED
@@ -43,20 +43,20 @@ class OpenAIGPT:
43
 
44
  def evaluate(self, prompt, text):
45
  model_name = self.config.scoring_model_name
46
- kwconfig = {"model": model_name,
47
  "prompt": f"<|endoftext|>{prompt}{text}",
48
  "max_tokens": 0, "echo": True, "logprobs": self.max_topk}
49
  # retry 1 time
50
  ntry = 2
51
  for idx in range(ntry):
52
  try:
53
- response = self.client.completions.create(**kwconfig)
54
  response = response.choices[0].logprobs
55
  return response
56
  break
57
  except Exception as e:
58
  if idx < ntry - 1:
59
- print(f'{model_name}, {kwconfig}: {e}. Retrying ...')
60
  time.sleep(5)
61
  continue
62
  raise e
 
43
 
44
  def evaluate(self, prompt, text):
45
  model_name = self.config.scoring_model_name
46
+ kwargs = {"model": model_name,
47
  "prompt": f"<|endoftext|>{prompt}{text}",
48
  "max_tokens": 0, "echo": True, "logprobs": self.max_topk}
49
  # retry 1 time
50
  ntry = 2
51
  for idx in range(ntry):
52
  try:
53
+ response = self.client.completions.create(**kwargs)
54
  response = response.choices[0].logprobs
55
  return response
56
  break
57
  except Exception as e:
58
  if idx < ntry - 1:
59
+ print(f'{model_name}, {kwargs}: {e}. Retrying ...')
60
  time.sleep(5)
61
  continue
62
  raise e