johann22 commited on
Commit
cf30049
1 Parent(s): 01a601e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ def run_gpt(
57
  print (seed)
58
  generate_kwargs = dict(
59
  temperature=1.0,
60
- max_new_tokens=1048,
61
  top_p=0.99,
62
  repetition_penalty=1.0,
63
  do_sample=True,
@@ -123,7 +123,7 @@ def call_main(purpose, task, history, directory, action_input):
123
  resp = run_gpt(
124
  ACTION_PROMPT,
125
  stop_tokens=["observation:", "task:"],
126
- max_tokens=1048,
127
  purpose=purpose,
128
  task=task,
129
  history=history,
 
57
  print (seed)
58
  generate_kwargs = dict(
59
  temperature=1.0,
60
+ max_new_tokens=2096,
61
  top_p=0.99,
62
  repetition_penalty=1.0,
63
  do_sample=True,
 
123
  resp = run_gpt(
124
  ACTION_PROMPT,
125
  stop_tokens=["observation:", "task:"],
126
+ max_tokens=2096,
127
  purpose=purpose,
128
  task=task,
129
  history=history,