DSXiangLi commited on
Commit
a536ff7
1 Parent(s): 13d0afb
Files changed (1) hide show
  1. ape/llm.py +1 -1
ape/llm.py CHANGED
@@ -24,7 +24,7 @@ class LLMGPT(object):
24
 
25
  def __init__(self, openai_key):
26
  self.gen_llm = ChatOpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
27
- self.eval_llm = OpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, logprobs=2, verbose=True)
28
  self.test_llm = OpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
29
  self.gen_chain = None
30
  self.eval_chain = None
 
24
 
25
  def __init__(self, openai_key):
26
  self.gen_llm = ChatOpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
27
+ self.eval_llm = OpenAI(openai_api_key=openai_key, max_tokens=0, temperature=0.7, echo=True, logprobs=1)
28
  self.test_llm = OpenAI(openai_api_key=openai_key, max_tokens=2000, temperature=0.7, verbose=True)
29
  self.gen_chain = None
30
  self.eval_chain = None