Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -34,8 +34,7 @@ client = InferenceClient(
|
|
34 |
|
35 |
|
36 |
VERBOSE = True
|
37 |
-
MAX_HISTORY =
|
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.
|
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:","
|
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,
|