johann22 commited on
Commit
0dd64fb
1 Parent(s): 4a988ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -61,6 +61,7 @@ def run_gpt(
61
  max_new_tokens=1048,
62
  top_p=0.99,
63
  repetition_penalty=1.0,
 
64
  do_sample=True,
65
  seed=seed,
66
  )
@@ -105,6 +106,7 @@ def compress_history(purpose, task, history, directory):
105
  def call_search(purpose, task, history, directory, action_input):
106
  print("CALLING SEARCH")
107
  try:
 
108
  if "http" in action_input:
109
  if "<" in action_input:
110
  action_input = action_input.strip("<")
 
61
  max_new_tokens=1048,
62
  top_p=0.99,
63
  repetition_penalty=1.0,
64
+ num_beams=4,
65
  do_sample=True,
66
  seed=seed,
67
  )
 
106
  def call_search(purpose, task, history, directory, action_input):
107
  print("CALLING SEARCH")
108
  try:
109
+
110
  if "http" in action_input:
111
  if "<" in action_input:
112
  action_input = action_input.strip("<")