Spaces:
Runtime error
Runtime error
DSXiangLi
commited on
Commit
•
a536ff7
1
Parent(s):
13d0afb
add
Browse files- 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=
|
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
|