Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def generate(message, history, system, max_tokens, temperature, top_p, top_k, pe
|
|
55 |
top_k=top_k,
|
56 |
repetition_penalty=penalty,
|
57 |
max_tokens=max_tokens,
|
58 |
-
|
59 |
)
|
60 |
# generate outputs
|
61 |
llm = LLM(model=model)
|
@@ -89,7 +89,14 @@ with gr.Blocks(css=css) as demo:
|
|
89 |
label="Max tokens",
|
90 |
render=False,
|
91 |
),
|
92 |
-
gr.Slider(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
93 |
gr.Slider(
|
94 |
minimum=0.1,
|
95 |
maximum=1.0,
|
|
|
55 |
top_k=top_k,
|
56 |
repetition_penalty=penalty,
|
57 |
max_tokens=max_tokens,
|
58 |
+
stop_token_ids=[151645,151643],
|
59 |
)
|
60 |
# generate outputs
|
61 |
llm = LLM(model=model)
|
|
|
89 |
label="Max tokens",
|
90 |
render=False,
|
91 |
),
|
92 |
+
gr.Slider(
|
93 |
+
minimum=0.1,
|
94 |
+
maximum=1.0,
|
95 |
+
value=0.7,
|
96 |
+
step=0.1,
|
97 |
+
label="Temperature",
|
98 |
+
render=False,
|
99 |
+
),
|
100 |
gr.Slider(
|
101 |
minimum=0.1,
|
102 |
maximum=1.0,
|