acecalisto3 commited on
Commit
ed877ce
·
verified ·
1 Parent(s): 66cb5d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -34,8 +34,7 @@ client = InferenceClient(
34
 
35
 
36
  VERBOSE = True
37
- MAX_HISTORY = 100
38
- MODEL = mistralai/Mixtral-8x7B-Instruct-v0.1"
39
 
40
 
41
  def format_prompt(message, history):
@@ -61,7 +60,7 @@ def run_gpt(
61
  temperature=1.0,
62
  max_new_tokens=2096,
63
  top_p=0.99,
64
- repetition_penalty=1.0,
65
  do_sample=True,
66
  seed=seed,
67
  )
@@ -124,7 +123,7 @@ def call_search(purpose, task, history, directory, action_input):
124
  def call_main(purpose, task, history, directory, action_input):
125
  resp = run_gpt(
126
  ACTION_PROMPT,
127
- stop_tokens=["observation:", "task:", "action:","thought:"],
128
  max_tokens=2096,
129
  purpose=purpose,
130
  task=task,
 
34
 
35
 
36
  VERBOSE = True
37
+ MAX_HISTORY = 125
 
38
 
39
 
40
  def format_prompt(message, history):
 
60
  temperature=1.0,
61
  max_new_tokens=2096,
62
  top_p=0.99,
63
+ repetition_penalty=1.7,
64
  do_sample=True,
65
  seed=seed,
66
  )
 
123
  def call_main(purpose, task, history, directory, action_input):
124
  resp = run_gpt(
125
  ACTION_PROMPT,
126
+ stop_tokens=["observation:", "task:", "action:","though:", "codegen:"],
127
  max_tokens=2096,
128
  purpose=purpose,
129
  task=task,